-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathtest_models.ps1
32 lines (28 loc) · 969 Bytes
/
test_models.ps1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
mkdir .\log
mkdir .\log\test
$models = @( `
"BEVNet-all\mixed",
"BEVNet-head\mixed",
"BEVNet-feet\mixed",
"BEVNet-no_attn\mixed",
"BEVNet-shared\mixed",
"CSRNet\CSRNet2BEV-finetune",
"DSSINet\DSSINet2BEV-finetune",
"faster_rcnn_r_50_fpn_3x\FasterRCNN2BEV-finetune",
"mask_rcnn_r_50_fpn_3x\MaskRCNN2BEV",
"ivnet\CC-Oracle",
"ivnet\Feet2BEV",
"ivnet\Head2BEV"
)
foreach ($model in $models) {
if (-not(Test-Path -Path "log\test\$model\test\model-output.h5" -PathType Leaf)) {
python src\test.py --task-option-file checkpoints\$model\option.yaml --use-gpus 0
}
python src\visualize_model_output.py `
--model-output-file log\test\$model\test\model-output.h5 -j 8
python src\run_metrics.py `
--task-option-file checkpoints\$model\option.yaml `
--model-output-file log\test\$model\test\model-output.h5 `
--output-csv log\test\metric_result.csv `
--use-gpu 0
}