def main():
if not os.path.exists(PATH_TO_STORE_DOWNLOADED_PHOTOS):
os.makedirs(PATH_TO_STORE_DOWNLOADED_PHOTOS)
coco = COCO(ANNOTATION_FILE)
personCategory = coco.getCatIds(catNms=['person'])
personImageIds = coco.getImgIds(catIds=personCategory)
personImageData = coco.loadImgs(personImageIds)
personImageDataIterator = iter(personImageData)
counter = downloader.init_counter(PATH_TO_STORE_DOWNLOADED_PHOTOS)
while counter <= COUNT_IMAGES_TO_DOWNLOAD:
try:
filename_with_path, url = get_filename_and_url(
personImageDataIterator)
success = downloader.download_non_already_existing_images(
filename_with_path, url)
if success:
counter += 1
print("Download #", counter)
except StopIteration:
print('Tried to download more photos than available, stopping.')
counter = COUNT_IMAGES_TO_DOWNLOAD + 1
pass
downloader.zip_images(PHOTO_ZIPFILE_NAME, PATH_TO_STORE_DOWNLOADED_PHOTOS)
def main():
if not os.path.exists(PATH_TO_STORE_DOWNLOADED_CARTOON_IMAGES):
os.makedirs(PATH_TO_STORE_DOWNLOADED_CARTOON_IMAGES)
with open(PATH_TO_SAFEBOORU_ALL_DATA_CSV) as raw_data:
raw_data.readline() # ignore header line by reading it
counter = downloader.init_counter(PATH_TO_STORE_DOWNLOADED_CARTOON_IMAGES)
while counter <= config.COUNT_IMAGES_TO_DOWNLOAD:
success = fetch_and_save_image(raw_data.readline())
if success:
counter += 1
print('Download #', counter)
downloader.zip_images(CARTOON_IMAGES_ZIPFILE_NAME, PATH_TO_STORE_DOWNLOADED_CARTOON_IMAGES)
def main():
if not os.path.exists(PATH_TO_STORED_CARTOON_IMAGES):
print("Can not smooth images, path does not exist: ",
PATH_TO_STORED_CARTOON_IMAGES)
if not os.path.exists(PATH_TO_STORE_SMOOTHED_IMAGES):
os.makedirs(PATH_TO_STORE_SMOOTHED_IMAGES)
for filename in os.listdir(PATH_TO_STORED_CARTOON_IMAGES):
cartoon_images_filename = PATH_TO_STORED_CARTOON_IMAGES + filename
smoothed_images_filename = PATH_TO_STORE_SMOOTHED_IMAGES + filename
if not os.path.exists(smoothed_images_filename):
edge_smoothing(cartoon_images_filename, smoothed_images_filename)
else:
print("Skipping file, already exists, ", cartoon_images_filename)
downloader.zip_images(SMOOTHED_IMAGES_ZIPFILE_NAME,
PATH_TO_STORE_SMOOTHED_IMAGES)