projectstreet-detection-docker

commit a4ba6718930097a1525f76e1d34a719f97fc168f
parent ad17748668744db36bd7a8815f1a47c8ac2b7fc3
Author: Marcel <MTRNord@users.noreply.github.com>
Date:   Sun,  9 Oct 2016 01:03:27 +0200

Update download-image-by-link.py
Diffstat:
M
scripts/download-image-by-link.py
 | 
4
+++-

1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/scripts/download-image-by-link.py b/scripts/download-image-by-link.py
@@ -10,6 +10,7 @@ def store_raw_images():
     neg_images_link = 'http://image-net.org/api/text/imagenet.synset.geturls?wnid=n00523513'   
     neg_image_urls = urllib.urlopen(neg_images_link).read().decode()
     pic_num = 1
+    fail_images = 0
     
     if not os.path.exists('neg'):
         os.makedirs('neg')
@@ -24,11 +25,12 @@ def store_raw_images():
                 resized_image = cv2.resize(img, (100, 100))
                 cv2.imwrite("neg/"+str(pic_num)+".jpg",resized_image)
             else:
-                print "invalid img"
+                fail_images += 1
             pic_num += 1
             
         except Exception as e:
             print str(e)
+    print fail_images + " invalid imgages"
             
 def store_raw_images2():
     neg_images_link = 'http://image-net.org/api/text/imagenet.synset.geturls?wnid=n07942152'