Adding in the Initial Image Gathering

2019-04-13 00:37:11 -05:00
parent 63540bf0ce
commit 7d0a16680a
3 changed files with 983 additions and 0 deletions
@@ -0,0 +1,16 @@
+import pandas as pd
+
+from google_images_download import google_images_download
+
+df = pd.read_csv("pokemon.csv")
+
+response = google_images_download.googleimagesdownload()
+
+for pokemon in df["identifier"][:251]:
+    absolute_image_paths = response.download(
+        {
+            "keywords": pokemon,
+            "limit": 250,
+            "chromedriver": "/usr/lib/chromium-browser/chromedriver"
+        }
+    )