Skip to content

Commit c44f437

Browse files
authored
Add DANet and FastSCNN
add training bash and log for danet and fastscnn
2 parents 9183b45 + 1857d4e commit c44f437

File tree

4 files changed

+5374
-0
lines changed

4 files changed

+5374
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#!/usr/bin/env bash
2+
3+
# cmd for training
4+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./scripts/segmentation/train.py \
5+
--dataset citys \
6+
--model danet \
7+
--backbone resnet101 \
8+
--syncbn \
9+
--ngpus 8 \
10+
--lr 0.01 \
11+
--epochs 240 \
12+
--base-size 2048 \
13+
--crop-size 768 \
14+
--workers 32 \
15+
16+
# cmd for evaluation
17+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./scripts/segmentation/test.py \
18+
--model-zoo danet_resnet101_citys \
19+
--dataset citys \
20+
--batch-size 8 \
21+
--ngpus 8 \
22+
--eval \
23+
--pretrained \
24+
--base-size 2048 \
25+
--crop-size 768
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#!/usr/bin/env bash
2+
3+
# cmd for training
4+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./scripts/segmentation/train.py \
5+
--dataset citys \
6+
--model danet \
7+
--backbone resnet50 \
8+
--syncbn \
9+
--ngpus 8 \
10+
--lr 0.01 \
11+
--epochs 240 \
12+
--base-size 2048 \
13+
--crop-size 768 \
14+
--workers 32 \
15+
16+
# cmd for evaluation
17+
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./scripts/segmentation/test.py \
18+
--model-zoo danet_resnet50_citys \
19+
--dataset citys \
20+
--batch-size 8 \
21+
--ngpus 8 \
22+
--eval \
23+
--pretrained \
24+
--base-size 2048 \
25+
--crop-size 768

0 commit comments

Comments
 (0)