Lucas Oskorep
19 lines
515 B
Python
19 lines
515 B
Python
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"
|
|
}
|
|
)
|
|
|
|
# TODO: Need to clean data up here.... really should be added to another class as well you lazy asshole
|