-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathespdnet_greenhouse_uest_outsource2.sh
37 lines (36 loc) · 1.23 KB
/
espdnet_greenhouse_uest_outsource2.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
mr=0.1
tgt_port=0.5
#for mr in 0.1 0.3 0.5; do
#for tgt_port in 0.5; do
CUDA_VISIBLE_DEVICES=0 python uest_seg.py \
--random-mirror \
--test-scale 1.0 \
--rm-prob \
--test-flipping \
--num-classes 5 \
--learning-rate 0.00001 \
--save /tmp/runs/results_segmentation/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_greenhouse_more.txt \
--data-tgt-test-list ./vision_datasets/greenhouse/val_greenhouse_more.txt \
--batch-size 20 \
--gpu 0 \
--model espdnetue \
--restore-from /tmp/runs/results_segmentation/model_espdnetue_camvid/s_2.0_sch_hybrid_loss_ce_res_480_sc_0.5_2.0_rgb/20200523-200621/espdnetue_2.0_480_best.pth \
--runs-root /tmp/runs/uest \
--power 0.0 \
--epr 5 \
--num-rounds 20 \
--outsource camvid \
--use-uncertainty true \
--use-nid true \
--nid-bin 64 \
--outsource-weights ./results_segmentation/model_espdnet_camvid/s_2.0_sch_hybrid_loss_ce_res_480_sc_0.5_2.0_rgb_/20200420-095339/espdnet_2.0_480_best.pth \
--use-depth true \
--trainable-fusion true
# done
#done
# --use-nid true \
# --nid-bin 256 \