国外游戏ui设计网站/谷歌网站推广优化

下载的图片为RGB类型，会自动resize到299乘以299
需要提供一份csv文件的路径以及输出的地址。
python download_images.py --input_file=INPUT_FILE --output_dir=IMAGES_DIR
csv文件的内容为该图片名称已经下载地址还有定点裁剪的四个坐标。

"""Script which downloads dataset images.
Usage:python download_images.py --input_file=INPUT_FILE --output_dir=IMAGES_DIR
where:INPUT_FILE is input csv file with dataset description, i.e. dev_dataset.csvIMAGES_DIR is output directory where all images should be downloaded
Example:# create directory for imagesmkdir images# download images declared in dev_dataset.csvpython download_images.py --input_file=dev_dataset.csv --output_dir=images
Dependencies:Python 2.7 or higher.Pillow library: https://python-pillow.org/
"""from __future__ import absolute_import
from __future__ import division
from __future__ import print_functionimport argparse
import csv
import os
import sys
from functools import partial
from io import BytesIO
import multiprocessing
from multiprocessing.dummy import Pool as ThreadPoolfrom PIL import Imagetry:from urllib.request import urlopen
except ImportError:from urllib2 import urlopendef parse_args():"""Parses command line arguments."""parser = argparse.ArgumentParser(description='Tool to download dataset images.')parser.add_argument('--input_file', required=True,help='Location of dataset.csv')parser.add_argument('--output_dir', required=True,help='Output path to download images')parser.add_argument('--threads', default=multiprocessing.cpu_count() + 1,help='Number of threads to use')args = parser.parse_args()return args.input_file, args.output_dir, int(args.threads)def get_image(row, output_dir):"""Downloads the image that corresponds to the given row.Prints a notification if the download fails."""if not download_image(image_id=row[0],url=row[1],x1=float(row[2]),y1=float(row[3]),x2=float(row[4]),y2=float(row[5]),output_dir=output_dir):print("Download failed: " + str(row[0]))def download_image(image_id, url, x1, y1, x2, y2, output_dir):"""Downloads one image, crops it, resizes it and saves it locally."""output_filename = os.path.join(output_dir, image_id + '.png')if os.path.exists(output_filename):# Don't download image if it's already therereturn Truetry:# Download imageurl_file = urlopen(url)if url_file.getcode() != 200:return Falseimage_buffer = url_file.read()# Crop, resize and save imageimage = Image.open(BytesIO(image_buffer)).convert('RGB')w = image.size[0]h = image.size[1]image = image.crop((int(x1 * w), int(y1 * h), int(x2 * w),int(y2 * h)))image = image.resize((299, 299), resample=Image.ANTIALIAS)image.save(output_filename)except IOError:return Falsereturn Truedef main():input_filename, output_dir, n_threads = parse_args()if not os.path.isdir(output_dir):print("Output directory {} does not exist".format(output_dir))sys.exit()with open(input_filename) as input_file:reader = csv.reader(input_file)header_row = next(reader)rows = list(reader)try:row_idx_image_id = header_row.index('ImageId')row_idx_url = header_row.index('URL')row_idx_x1 = header_row.index('x1')row_idx_y1 = header_row.index('y1')row_idx_x2 = header_row.index('x2')row_idx_y2 = header_row.index('y2')except ValueError as e:print('One of the columns was not found in the source file: ',e.message)rows = [(row[row_idx_image_id], row[row_idx_url], float(row[row_idx_x1]),float(row[row_idx_y1]), float(row[row_idx_x2]),float(row[row_idx_y2])) for row in rows]if n_threads > 1:pool = ThreadPool(n_threads)partial_get_images = partial(get_image, output_dir=output_dir)for i, _ in enumerate(pool.imap_unordered(partial_get_images, rows),1):sys.stderr.write('\rDownloaded {0} images'.format(i + 1))pool.close()pool.join()else:failed_to_download = set()for idx in range(len(rows)):row = rows[idx]if not download_image(image_id=row[0],url=row[1],x1=float(row[2]),y1=float(row[3]),x2=float(row[4]),y2=float(row[5]),output_dir=output_dir):failed_to_download.add(row[row_idx_image_id])sys.stdout.write('\rDownloaded {0} images'.format(idx + 1))sys.stdout.flush()print()if failed_to_download:print('\nUnable to download images with the following IDs:')for image_id in failed_to_download:print(image_id)if __name__ == '__main__':main()