Updating the input image sizes to have a higher resolution. Currently unclean data is reaching 70% accuracy.

2019-04-14 18:50:53 -05:00
parent aabd88d4fb
commit de117a55ae
23 changed files with 1869 additions and 7 deletions
@@ -0,0 +1,20 @@
+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 ["abra", "xatu", "yanma", "zapdos", "zubat"]:  # df["identifier"][:251]:
+    absolute_image_paths = response.download(
+        {
+            "keywords": pokemon,
+            "limit": 250,
+            "chromedriver": "/usr/lib/chromium-browser/chromedriver",
+            # This needs to be changed based on the computer trying to download the images
+            "format": "jpg"
+        }
+    )
+
+# TODO: Need to clean data up here.... really should be added to another class as well you lazy asshole