Update detect.py

This generator causes an error in detect_windows because it is run twice there. Better store it as a data structure instead.
egalev · Oct 2, 2014 · 4e0868d · 4e0868d
1 parent 2fa2852
commit 4e0868d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/detect.py b/python/detect.py
@@ -130,10 +130,10 @@ def main(argv):
     # Detect.
     if args.crop_mode == 'list':
         # Unpack sequence of (image filename, windows).
-        images_windows = (
+        images_windows = [
             (ix, inputs.iloc[np.where(inputs.index == ix)][COORD_COLS].values)
             for ix in inputs.index.unique()
-        )
+        ]
         detections = detector.detect_windows(images_windows)
     else:
         detections = detector.detect_selective_search(inputs)