20 lines
493 B
PowerShell
20 lines
493 B
PowerShell
$ErrorActionPreference = "Stop"
|
|
|
|
$repo = $PSScriptRoot
|
|
$env:PYTHONPATH = "$repo\.pydeps"
|
|
$env:YOLO_CONFIG_DIR = "$repo\.ultralytics"
|
|
|
|
Set-Location $repo
|
|
|
|
py -3.11 .\07_build_public_shoe_dataset.py --clean
|
|
|
|
py -3.11 .\08_train_compare_models.py `
|
|
--data "$repo\datasets\shoe-public-mix\data.yaml" `
|
|
--roi-dir "$repo\datasets\roi-shoes" `
|
|
--project "$repo\runs\shoe_compare" `
|
|
--models "yolov8s.pt" "yolo11s.pt" "yolo26s.pt" `
|
|
--epochs 40 `
|
|
--imgsz 640 `
|
|
--batch 16 `
|
|
--device 0
|