46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
# %%
|
||
|
|
# download src_ref_files from Zenodo
|
||
|
|
import os
|
||
|
|
import requests
|
||
|
|
|
||
|
|
url = 'https://zenodo.org/records/14092478/files/src_rec_file.dat?download=1'
|
||
|
|
|
||
|
|
path = "1_src_rec_files/src_rec_file.dat"
|
||
|
|
|
||
|
|
# check file existence
|
||
|
|
if not os.path.exists(path):
|
||
|
|
try:
|
||
|
|
os.mkdir("1_src_rec_files")
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
print("Downloading src_rec_file.dat from Zenodo...")
|
||
|
|
response = requests.get(url, stream=True)
|
||
|
|
with open(path, 'wb') as out_file:
|
||
|
|
out_file.write(response.content)
|
||
|
|
print("Download complete.")
|
||
|
|
else:
|
||
|
|
print("src_rec_file.dat already exists.")
|
||
|
|
|
||
|
|
# %%
|
||
|
|
# download initial model from Zenodo
|
||
|
|
|
||
|
|
url = 'https://zenodo.org/records/14092478/files/model_init_N121_121_121.h5?download=1'
|
||
|
|
|
||
|
|
path = "2_models/model_init_N121_121_121.h5"
|
||
|
|
|
||
|
|
# check file existence
|
||
|
|
if not os.path.exists(path):
|
||
|
|
try:
|
||
|
|
os.mkdir("2_models")
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
print("Downloading model_init_N121_121_121.h5 from Zenodo...")
|
||
|
|
response = requests.get(url, stream=True)
|
||
|
|
with open(path, 'wb') as out_file:
|
||
|
|
out_file.write(response.content)
|
||
|
|
print("Download complete.")
|
||
|
|
else:
|
||
|
|
print("model_init_N121_121_121.h5 already exists.")
|
||
|
|
|
||
|
|
|