Merge pull request thtrieu#581 from off99555/hotfix

Show error message when an image is corrupted
LittleD3092 · Feb 26, 2018 · 5fe906b · 5fe906b
2 parents 17d56e4 + d79d08b
commit 5fe906b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/darkflow/net/yolo/data.py b/darkflow/net/yolo/data.py
@@ -110,7 +110,13 @@ def shuffle(self):
 
             for j in range(b*batch, b*batch+batch):
                 train_instance = data[shuffle_idx[j]]
-                inp, new_feed = self._batch(train_instance)
+                try:
+                    inp, new_feed = self._batch(train_instance)
+                except ZeroDivisionError:
+                    print("This image's width or height are zeros: ", train_instance[0])
+                    print('train_instance:', train_instance)
+                    print('Please remove or fix it then try again.')
+                    raise
 
                 if inp is None: continue
                 x_batch += [np.expand_dims(inp, 0)]