Skip to content
Commits on Source (2)
colorizers/siggraph17-df00044c.pth
colorizers/colorization_release_v2-9b330a0b.pth
colorizers/col_new.pth
__pycache__
*.png
\ No newline at end of file
*.png
.ipynb_checkpoints
import requests
import json
import threading
import os
def get_json_by_url(url):
res = requests.get(url).content
return json.loads(res)
def get_manifesturls_from_collectionurl(url):
x = get_json_by_url(url)
return [el['@id'] for el in x['manifests']]
def get_imgurls_from_manifesturl(url):
x = get_json_by_url(url)
res = []
for s in x['sequences']:
for c in s['canvases']:
for i in c['images']:
res.append(i['resource']['@id'])
return res
def get_imgurls_from_collectionurl(url):
res = []
for manifest_url in get_manifesturls_from_collectionurl(url):
tmp = get_imgurls_from_manifesturl(manifest_url)
for el in tmp:
res.append(el)
return res
def download_images(img_urls, img_paths):
for u, n in zip(img_urls, img_paths):
img_bytes = requests.get(u).content
with open(n, '+wb') as img_file:
img_file.write(img_bytes)
def download_images_multithreded(img_urls, img_paths, nb_threads=5):
chunk_idcs = list(range(0, len(img_urls), int(len(img_urls)/nb_threads))) + [len(img_urls)]
chunks_urls = [img_urls[i:j] for i,j in zip(chunk_idcs[:-1], chunk_idcs[1:])]
chunks_paths = [img_paths[i:j] for i,j in zip(chunk_idcs[:-1], chunk_idcs[1:])]
for urls, paths in zip(chunks_urls, chunks_paths):
threading.Thread(target=download_images, args=[urls, paths]).start()
def create_paths_from_iiifurls(img_urls, base_path=''):
res = [el.split('images/')[1] for el in img_urls]
res = ['_'.join(el.split('/')[:-4]) for el in res]
res = [el+'.'+u.split('.')[-1] for el, u in zip(res, img_urls)]
res = [os.path.join(base_path, el) for el in res]
return res
This diff is collapsed.
# Seems not to work for Binder
#wget -q -o colorizers/colorization_release_v2-9b330a0b.pth https://colorizers.s3.us-east-2.amazonaws.com/colorization_release_v2-9b330a0b.pth
#wget -q -o colorizers/siggraph17-df00044c.pth https://colorizers.s3.us-east-2.amazonaws.com/siggraph17-df00044c.pth
\ No newline at end of file
This diff is collapsed.