Files
ICE_Dataset/scripts/orchestrator_322k.ps1
kirl a64690477a Dataset: add READMEs, organize scripts, update .gitignore
- Add README.md to all 16 dataset directories with script references
- Centralize generation scripts under scripts/ with full README
- Update .gitignore to cover all data directories, keep only metadata
- Remove redundant CCPD/.gitignore (parent covers same rules)
- Remove stale scripts/test/ (254MB pipeline residue)
- Fix label parser reference: scripts are authoritative source of truth
2026-06-23 14:41:39 +08:00

39 lines
1.6 KiB
PowerShell

# 322K 全自动: 等生成完 → 构建 → EXP_CUT → TRANS → 训练
# 用法: 生成进程跑完后执行此脚本
$ErrorActionPreference = "Continue"
$PY = "C:/Program Files/Python313/python.exe"
$ICE = "C:\Users\Kirl\Desktop\ICEv2\ICE_Dataset"
Write-Host "===== 322K Pipeline =====" -ForegroundColor Cyan
Write-Host "Waiting for SPE + Random generation to complete..."
Write-Host "SPE dir: $ICE\CCPD_SPE\ccpd_base"
Write-Host "Random dir: $ICE\CCPD_Random_Blue150K_Green100K"
# Wait for SPE (9 types x 8000)
$spe_types = @('police','army','gangao','shiguan_old','shiguan_new','lingguan_old','lingguan_new','coach','new_energy')
do {
$done = $true
foreach ($t in $spe_types) {
$c = @(Get-ChildItem "$ICE\CCPD_SPE\ccpd_base\$t" -Filter *.jpg -ErrorAction SilentlyContinue).Count
if ($c -lt 8000) { $done = $false; break }
}
if (-not $done) { Write-Host " SPE not ready, waiting 60s..."; Start-Sleep 60 }
} while (-not $done)
Write-Host "SPE: OK" -ForegroundColor Green
# Wait for Random
do {
$b = @(Get-ChildItem "$ICE\CCPD_Random_Blue150K_Green100K\blue" -Filter *.jpg -ErrorAction SilentlyContinue).Count
$g = @(Get-ChildItem "$ICE\CCPD_Random_Blue150K_Green100K\green" -Filter *.jpg -ErrorAction SilentlyContinue).Count
if ($b -lt 150000 -or $g -lt 100000) {
Write-Host " Random: blue=$b green=$g, waiting 60s..."
Start-Sleep 60
} else { break }
} while ($true)
Write-Host "Random: blue=$b green=$g OK" -ForegroundColor Green
# Now run the pipeline
Write-Host "`nAll data ready. Starting pipeline..." -ForegroundColor Cyan
& "C:/Program Files/Python313/python.exe" "$PSScriptRoot\pipeline_322k.py"