|
import requests |
|
|
|
model_urls = { |
|
"convnext_tiny_1k": "https://dl.fbaipublicfiles.com/convnext/convnext_tiny_1k_224_ema.pth", |
|
"convnext_small_1k": "https://dl.fbaipublicfiles.com/convnext/convnext_small_1k_224_ema.pth", |
|
"convnext_base_1k": "https://dl.fbaipublicfiles.com/convnext/convnext_base_1k_224_ema.pth", |
|
"convnext_large_1k": "https://dl.fbaipublicfiles.com/convnext/convnext_large_1k_224_ema.pth", |
|
"convnext_tiny_22k": "https://dl.fbaipublicfiles.com/convnext/convnext_tiny_22k_224.pth", |
|
"convnext_small_22k": "https://dl.fbaipublicfiles.com/convnext/convnext_small_22k_224.pth", |
|
"convnext_base_22k": "https://dl.fbaipublicfiles.com/convnext/convnext_base_22k_224.pth", |
|
"convnext_large_22k": "https://dl.fbaipublicfiles.com/convnext/convnext_large_22k_224.pth", |
|
"convnext_xlarge_22k": "https://dl.fbaipublicfiles.com/convnext/convnext_xlarge_22k_224.pth", |
|
} |
|
|
|
|
|
|
|
for model_name, url in model_urls.items(): |
|
response = requests.get(url) |
|
with open(f"{model_name}.pth", "wb") as f: |
|
f.write(response.content) |
|
|
|
print(f"{model_name} downloaded!") |
|
|
|
|
|
|
|
FID_WEIGHTS_URL = ('https://github.com/w86763777/pytorch-gan-metrics/releases/' |
|
'download/v0.1.0/pt_inception-2015-12-05-6726825d.pth') |
