Skip to content

Commit 2717c9b

Browse files
committed
add customer dataset and associated config file
1 parent e0323e3 commit 2717c9b

File tree

3 files changed

+24
-15
lines changed

3 files changed

+24
-15
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ frankfurt_000001_079206_leftImg8bit.png,frankfurt_000001_079206_gtFine_labelIds.
109109
...
110110
```
111111
Each line is a pair of training sample and ground truth image path, which are separated by a single comma `,`.
112-
Then you need to change the field of `im_root` and `train/val_im_anns` in the configuration files. If you found what shows in `cityscapes_cv2.py` is not clear, you can also see `coco.py`.
112+
Then you need to change the field of `im_root` and `train/val_im_anns` in the config file. I prepared a demo config file for you named [`bisenet_customer.py`](./configs/bisenet_customer.py). You can start from this conig file.
113113

114114

115115
## train

configs/bisenet_customer.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
2+
cfg = dict(
3+
model_type='bisenetv1',
4+
n_cats=20,
5+
num_aux_heads=2,
6+
lr_start=1e-2,
7+
weight_decay=5e-4,
8+
warmup_iters=1000,
9+
max_iter=80000,
10+
dataset='CustomerDataset',
11+
im_root='./datasets/cityscapes',
12+
train_im_anns='./datasets/cityscapes/train.txt',
13+
val_im_anns='./datasets/cityscapes/val.txt',
14+
scales=[0.75, 2.],
15+
cropsize=[512, 512],
16+
eval_crop=[512, 512],
17+
eval_scales=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75],
18+
ims_per_gpu=8,
19+
eval_ims_per_gpu=2,
20+
use_fp16=True,
21+
use_sync_bn=False,
22+
respth='./res',
23+
)

run.sh

Lines changed: 0 additions & 14 deletions
This file was deleted.

0 commit comments

Comments
 (0)