Add FRRN and FCN configs with new dataloader interface

meetps · Jan 8, 2019 · 4654882 · 4654882
1 parent e555b39
commit 4654882
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/configs/fcn8s_pascal.yml b/configs/fcn8s_pascal.yml
@@ -7,6 +7,7 @@ data:
     img_rows: 'same'
     img_cols: 'same'
     path: /private/home/meetshah/datasets/VOC/060817/VOCdevkit/VOC2012/
+    sbd_path: /private/home/meetshah/datasets/VOC/benchmark_RELEASE/
 training:
     train_iters: 300000
     batch_size: 1

diff --git a/configs/frrnB_cityscapes.yml b/configs/frrnB_cityscapes.yml
@@ -0,0 +1,22 @@
+model:
+    arch: frrnB
+data:
+    dataset: cityscapes
+    train_split: train
+    val_split: val
+    img_rows: 512
+    img_cols: 1024
+    path: /private/home/meetshah/misc_code/ps/data/VOCdevkit/VOC2012/
+training:
+    train_iters: 85000
+    batch_size: 2
+    val_interval: 500
+    print_interval: 25
+    optimizer:
+        lr: 1.0e-4
+    l_rate: 1.0e-4
+    l_schedule:
+    momentum: 0.99
+    weight_decay: 0.0005
+    resume: frrnB_cityscapes_best_model.pkl
+    visdom: False