Skip broken images in avatars/ folder

JohanAR · JohanAR · commit 95071c560649 · 2021-01-13T11:12:37.000+01:00
diff --git a/afy/cam_fomm.py b/afy/cam_fomm.py
@@ -74,6 +74,10 @@ def load_images(IMG_SIZE = 256):
     for i, f in enumerate(images_list):
         if f.endswith('.jpg') or f.endswith('.jpeg') or f.endswith('.png'):
             img = cv2.imread(f)
+            if img is None:
+                log("Failed to open image: {}".format(f))
+                continue
+
             if img.ndim == 2:
                 img = np.tile(img[..., None], [1, 1, 3])
             img = img[..., :3][..., ::-1]
