Commit 3251c773 authored by tim.scherr's avatar tim.scherr
Browse files

model and evaluation software download bugfix

parent f1d87376
......@@ -61,7 +61,7 @@ if __name__ == "__main__":
os.remove(challengedata_path / "{}.zip".format(cell_type))
# Download evaluation software
if len(list(evalsoftware_path.glob('*'))) == 0:
if len(list(evalsoftware_path.glob('*'))) <= 1:
print('Downloading evaluation software ...')
download_data(url=evalsoftware_url, target=evalsoftware_path)
......
......@@ -24,7 +24,7 @@ if __name__ == "__main__":
model_url = 'http://public.celltrackingchallenge.net/participants/KIT-Sch-GE%20(2).zip'
# Download training set
if len(list(model_path.glob('*'))) == 0:
if len(list(model_path.glob('*.pth'))) == 0:
print('Downloading models ...')
download_data(url=model_url, target=model_path)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment