Spaces:
Sleeping
Sleeping
File size: 1,412 Bytes
8a6df40 e67eb77 8a6df40 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import requests
import json
import posixpath
import os
from posenet import MOBILENET_V1_CHECKPOINTS
GOOGLE_CLOUD_STORAGE_DIR = 'https://storage.googleapis.com/tfjs-models/weights/posenet/'
def download_json(checkpoint, filename, base_dir):
url = posixpath.join(GOOGLE_CLOUD_STORAGE_DIR, checkpoint, filename)
response = requests.get(url)
data = json.loads(response.content)
with open(os.path.join(base_dir, checkpoint, filename), 'w') as outfile:
json.dump(data, outfile)
def download_file(checkpoint, filename, base_dir):
url = posixpath.join(GOOGLE_CLOUD_STORAGE_DIR, checkpoint, filename)
response = requests.get(url)
f = open(os.path.join(base_dir, checkpoint, filename), 'wb')
f.write(response.content)
f.close()
def download(checkpoint, base_dir='TryYours-Virtual-Try-On/posenet/converter/weights/'):
save_dir = os.path.join(base_dir, checkpoint)
if not os.path.exists(save_dir):
os.makedirs(save_dir)
download_json(checkpoint, 'manifest.json', base_dir)
f = open(os.path.join(save_dir, 'manifest.json'), 'r')
json_dict = json.load(f)
for x in json_dict:
filename = json_dict[x]['filename']
print('Downloading', filename)
download_file(checkpoint, filename, base_dir)
def main():
checkpoint = MOBILENET_V1_CHECKPOINTS[101]
download(checkpoint)
if __name__ == "__main__":
main()
|