Update fcn.py

Tramac · Oct 28, 2020 · 82db70c · 82db70c
1 parent 61626f2
commit 82db70c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/core/models/fcn.py b/core/models/fcn.py
@@ -156,7 +156,7 @@ def get_fcn32s(dataset='pascal_voc', backbone='vgg16', pretrained=False, root='~
         'ade20k': 'ade',
         'coco': 'coco',
         'citys': 'citys',
-        'sbu': 'sub',
+        'sbu': 'sbu',
     }
     from ..data.dataloader import datasets
     model = FCN32s(datasets[dataset].NUM_CLASS, backbone=backbone, pretrained_base=pretrained_base, **kwargs)
@@ -176,7 +176,7 @@ def get_fcn16s(dataset='pascal_voc', backbone='vgg16', pretrained=False, root='~
         'ade20k': 'ade',
         'coco': 'coco',
         'citys': 'citys',
-        'sbu': 'sub',
+        'sbu': 'sbu',
     }
     from ..data.dataloader import datasets
     model = FCN16s(datasets[dataset].NUM_CLASS, backbone=backbone, pretrained_base=pretrained_base, **kwargs)
@@ -196,7 +196,7 @@ def get_fcn8s(dataset='pascal_voc', backbone='vgg16', pretrained=False, root='~/
         'ade20k': 'ade',
         'coco': 'coco',
         'citys': 'citys',
-        'sbu': 'sub',
+        'sbu': 'sbu',
     }
     from ..data.dataloader import datasets
     model = FCN8s(datasets[dataset].NUM_CLASS, backbone=backbone, pretrained_base=pretrained_base, **kwargs)