projectstreet-detection-docker

commit ad0565247d14d566804963a6dfc3a411793ca901
parent 8fa09a909fd3a817807b1ac290e3f2bbadbc2b08
Author: Marcel <MTRNord@users.noreply.github.com>
Date:   Fri, 14 Oct 2016 13:10:07 +0200

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

1 file changed, 12 insertions(+), 4 deletions(-)
diff --git a/scripts/download-image-by-link.py b/scripts/download-image-by-link.py
@@ -7,6 +7,10 @@ import cv2
 import numpy as np
 import os
 from socket import timeout
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument('--stage', help='Stage to run')
 
 def store_raw_images():
     neg_images_link = 'http://image-net.org/api/text/imagenet.synset.geturls?wnid=n00523513'   
@@ -130,7 +134,11 @@ def create_pos_n():
                 with open('bg.txt','a') as f:
                     f.write(line)
                     
-store_raw_images()
-store_raw_images2()
-store_raw_pos_images()
-create_pos_n()
+
+if args.stage == "1_1":
+    store_raw_images()
+if args.stage == "1_2":
+    store_raw_images2()
+if args.stage == "1_3":
+    store_raw_pos_images()
+    create_pos_n()