You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.3 KiB
43 lines
1.3 KiB
import requests
|
|
import os
|
|
import tarfile
|
|
import sys
|
|
|
|
urls = [
|
|
"https://s3-ap-northeast-1.amazonaws.com/mycityreport/trainedModels.tar.gz",
|
|
"http://download.tensorflow.org/models/object_detection/rfcn_resnet101_coco_2018_01_28.tar.gz",
|
|
"http://download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_kitti_2018_01_28.tar.gz"
|
|
]
|
|
|
|
paths = ["../server_side/api/modules", ".", '../server_side/api/modules']
|
|
|
|
for i in range(len(urls)):
|
|
if i == 0:
|
|
continue
|
|
url = urls[i]
|
|
print("[INFO]: Downloadinng file: {} to temp.tar.gz!".format(url.split("/")[-1]))
|
|
|
|
with open("temp.tar.gz", 'wb') as f:
|
|
response = requests.get(url, stream=True)
|
|
total = response.headers.get('content-length')
|
|
|
|
if total is None:
|
|
f.write(response.content)
|
|
else:
|
|
downloaded = 0
|
|
total = int(total)
|
|
for data in response.iter_content(chunk_size=max(int(total/1000), 1024*1024)):
|
|
downloaded += len(data)
|
|
f.write(data)
|
|
done = int(50*downloaded/total)
|
|
sys.stdout.write('\r[{}{}]'.format('*' * done, '.' * (50-done)))
|
|
sys.stdout.flush()
|
|
sys.stdout.write('\n')
|
|
print("[INFO]: Done downloading, now extracting!")
|
|
tar = tarfile.open("temp.tar.gz")
|
|
tar.extractall(path=paths[i])
|
|
tar.close()
|
|
os.remove("temp.tar.gz")
|
|
print("[INFO]: Done downloading, now deleting!")
|
|
print("[INFO]: Done")
|
|
|