| 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) | |