AdGPT / image_download.py
goodmodeler's picture
update
c269ab0
raw
history blame
953 Bytes
import flickrapi
import requests
import os
# Your Flickr API credentials
FLICKR_PUBLIC = '0ff89a88a2a61c24f452774ad32ee62c'
FLICKR_SECRET = '35c5034466630c82'
# Create Flickr API object
flickr = flickrapi.FlickrAPI(FLICKR_PUBLIC, FLICKR_SECRET, format='parsed-json')
# Search for images with relevant tags
results = flickr.photos.search(
text='advertisement',
per_page=50,
media='photos',
sort='relevance',
extras='url_o,url_l,url_c,tags',
content_type=1,
safe_search=1
)
photos = results['photos']['photo']
# Create folder to save images
# os.makedirs('flickr_brooklyn_ads', exist_ok=True)
# Download images
for i, photo in enumerate(photos):
url = photo.get('url_o') or photo.get('url_l') or photo.get('url_c')
if url:
img_data = requests.get(url).content
with open(f'nyc_ads_dataset/img_{i}.jpg', 'wb') as handler:
handler.write(img_data)
print(f"Downloaded: img_{i}.jpg")