added instructions for scoring single image

malteprinzler · Nov 17, 2020 · abe0740 · abe0740
1 parent 0731462
commit abe0740
Show file tree

Hide file tree

Showing 2 changed files with 97 additions and 25 deletions.
diff --git a/notebooks/attribute_prediction.ipynb b/notebooks/attribute_prediction.ipynb
diff --git a/python_code/iMatDataset.py b/python_code/iMatDataset.py
@@ -184,8 +184,11 @@ def __init__(self, labels_file, img_dir, attr_descr_file, image_augmentations=li
         self.img_dir = img_dir
 
         # getting labels
-        with open(labels_file, "r") as f:
-            self.labels = json.load(f)["annotations"]
+        if labels_file and os.path.exists(labels_file):
+            with open(labels_file, "r") as f:
+                self.labels = json.load(f)["annotations"]
+        else:
+            self.labels = []
 
         ###
         # Transformations
@@ -206,7 +209,8 @@ def __init__(self, labels_file, img_dir, attr_descr_file, image_augmentations=li
 
         self.attr_descr = self.get_attribute_description(attr_descr_file)
         self.n_attrs = len(self.attr_descr)
-        self.len = (length if length is not None else min(len(self.labels), len(os.listdir(self.img_dir))))
+        self.len = (length if length is not None else min(len(self.labels),
+                                                          len(os.listdir(self.img_dir)) if self.img_dir else 0))
 
     def __getitem__(self, item):
         """