-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathespdnet_uest_23.sh
48 lines (46 loc) · 1.77 KB
/
espdnet_uest_23.sh
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
train_list="train_2_3.lst"
val_list="val_greenhouse_more.lst"
restore_from="/tmp/runs/model_espdnetue_camvid/s_2.0_sch_hybrid_loss_ce_res_480_sc_0.5_2.0_rgb/20200715-230509/espdnetue_2.0_480_best.pth"
outsource1=camvid
outsource2=cityscapes
outsource3=forest
os_model1=espdnetue
os_model2=espdnetue
os_model3=espdnetue
os_weights1="/tmp/runs/model_espdnetue_camvid/s_2.0_sch_hybrid_loss_ce_res_480_sc_1.0_1.0_rgb/20200715-164256/espdnetue_2.0_480_best.pth"
os_weights2="/tmp/runs/model_espdnetue_city/s_2.0_sch_hybrid_loss_ce_res_512_sc_0.25_0.5_rgb/20200717-161219/espdnetue_2.0_512_best.pth"
os_weights3="/tmp/runs/model_espdnetue_forest/s_2.0_sch_hybrid_loss_ce_res_480_sc_1.0_1.0_rgb/20200729-181614/espdnetue_2.0_480_best.pth"
# Multi-source (Camvid-Cityscapes-forest)
CUDA_VISIBLE_DEVICES=0 python uest_seg_multi_os.py \
--random-mirror \
--test-scale 1.0 \
--rm-prob \
--test-flipping \
--num-classes 5 \
--learning-rate 0.00001 \
--save /tmp/runs/uest \
--data-path ./vision_datasets/ \
--data-src greenhouse \
--data-src-list ./vision_datasets/camvid/train_camvid.txt \
--data-tgt-train-list ./vision_datasets/greenhouse/$train_list \
--data-tgt-test-list ./vision_datasets/greenhouse/$val_list \
--batch-size 24 \
--gpu 0 \
--model espdnetue \
--restore-from $restore_from \
--runs-root /tmp/runs/uest \
--epr 5 \
--num-rounds 10 \
--use-uncertainty true \
--conservative-label true \
--label-update 3 \
--merge-label-policy all \
--os-model1 $os_model1 \
--outsource1 $outsource1\
--os-weights1 $os_weights1 \
--os-model2 $os_model2 \
--outsource2 $outsource2 \
--os-weights2 $os_weights2 \
--os-model3 $os_model3 \
--outsource3 $outsource3 \
--os-weights3 $os_weights3