|
|
| import os |
| import requests, zipfile, io |
|
|
| IMAGE_DIR = 'flickr30k_images/flickr30k_images' |
|
|
| def downlad_images() : |
|
|
| zip_file = 'data/flickr.zip' |
|
|
| |
| zip_file_url = 'https://drive.google.com/open?id=14QhofCbby053kWbVeWEBHCxOROQS-bjN&authuser=0' |
|
|
| try : |
| if not os.path.exists(IMAGE_DIR) : |
| if not os.path.exists(zip_file) : |
| r = requests.get(zip_file_url) |
| z = zipfile.ZipFile(io.BytesIO(r.content)) |
| z.extractall(".") |
|
|
| else : |
| with zipfile.ZipFile(zip_file, 'r') as zip_ref: |
| zip_ref.extractall(".") |
| except : |
| print("Problems with image file download") |
|
|
| return |
