fix bug getting imagenet data (#52)

SeldonIO · May 2, 2019 · 339cd3e · 339cd3e
1 parent 75aba9b
commit 339cd3e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/alibi/datasets.py b/alibi/datasets.py
@@ -59,6 +59,8 @@ def imagenet(category: str = 'Persian cat', nb_images: int = 10, target_size: tu
         image = np.asarray(bytearray(resp.read()), dtype="uint8")
         resp.close()
         image = cv2.imdecode(image, cv2.IMREAD_COLOR)
+        if image is None:
+            continue
         image = np.expand_dims(cv2.resize(image, target_size), axis=0)
         if np.std(image) < min_std:  # do not include empty images
             continue

diff --git a/examples/anchor_image_imagenet.ipynb b/examples/anchor_image_imagenet.ipynb