- put camvid in __init__ again (it was fixed in a previous commit)

- remove bsd500 from this branch (has neurofire dependency)
inferno-pytorch · Jun 18, 2018 · c70aee0 · c70aee0
1 parent fa4bcc6
commit c70aee0
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 215 deletions.
diff --git a/inferno/io/box/__init__.py b/inferno/io/box/__init__.py
@@ -1,7 +1,5 @@
 """Things that work out of the box. ;)"""
 
-# FIXME Camvid requires some deprecated torch function
-# from .camvid import CamVid, get_camvid_loaders
+from .camvid import CamVid, get_camvid_loaders
 from .cityscapes import Cityscapes, get_cityscapes_loaders
 from .cifar import get_cifar10_loaders, get_cifar100_loaders
-from .bsd500 import BSD500, get_bsd500_loaders, get_bsd500_dataset
diff --git a/inferno/io/box/bsd500.py b/inferno/io/box/bsd500.py
diff --git a/inferno/io/box/camvid.py b/inferno/io/box/camvid.py
@@ -164,6 +164,7 @@ def download(self):
         raise NotImplementedError
 
 
+# noinspection PyTypeChecker
 def get_camvid_loaders(root_directory, image_shape=(360, 480), labels_as_onehot=False,
                        train_batch_size=1, validate_batch_size=1, test_batch_size=1,
                        num_workers=2):