upload test

downloader.py

@@ -0,0 +1,27 @@
+import pandas as pd
+import requests
+import os
+from PIL import Image
+from io import BytesIO
+
+df = pd.read_table("../stock_each_shot_code.csv")
+
+category_count = {}
+for record in df.to_records():
+    url = record[2]
+    category = record[3]
+    if category not in os.listdir():
+        os.mkdir(category)
+    if category not in category_count:
+        category_count[category] = 1
+    image_name = str(category_count[category]) + ".jpg"
+    if image_name not in os.listdir(category):
+        try:
+            response = requests.get(url)
+            img = Image.open(BytesIO(response.content)).convert('RGB')
+            img.save(category + "/" + image_name)
+            category_count[category] += 1
+        except Exception:
+            print("----------")
+            print(category)
+            print(url)