def download_dataset(download_dir: str):
"""
This script downloads the MIDV-500 dataset and unzips the folders.
"""
for k, v in links_set.items():
dst = os.path.join(download_dir, k)
for link in v:
print(
"--------------------------------------------------------------"
)
# doownlaod zip file
filename = link.split("/")[-1]
print("\nDownloading:", filename)
download(link, dst)
print("Downloaded:", filename)
# unzip zip file
print("Unzipping:", filename)
zip_path = os.path.join(dst, filename)
if k == "midv2019":
dst_new = os.path.join(dst, filename.replace(".zip", ""))
unzip(zip_path, dst_new)
else:
unzip(zip_path, dst)
print("Unzipped:", filename.replace(".zip", ""))
# remove zip file
os.remove(zip_path)
def test_unzip(self):
from midv500.utils import unzip
# check if annotations paths can be read
file_path = "tests/test_data/CA43_01.zip"
dest_dir = "tests/test_data/"
# read annotations paths
unzip(file_path, dest_dir)
def download_dataset(download_dir: str, dataset_name: str = "midv500"):
"""
This script downloads the MIDV-500 dataset with extra files and unzips the folders.
dataset_name: str
"midv500": https://doi.org/10.18287/2412-6179-2019-43-5-818-824
"midv2019": https://doi.org/10.1117/12.2558438
"all": midv500 + midv2019
"""
if dataset_name == "midv500":
links_set = {
"midv500": midv500_links,
}
elif dataset_name == "midv2019":
links_set = {
"midv2019": midv2019_links,
}
elif dataset_name == "all":
links_set = {
"midv500": midv500_links,
"midv2019": midv2019_links,
}
else:
Exception(
'Invalid dataset_name, try one of "midv500", "midv2019" or "all".')
for k, v in links_set.items():
dst = os.path.join(download_dir, k)
for link in v:
print(
"--------------------------------------------------------------"
)
# download zip file
link = link.replace("\\", "/") # for windows
filename = link.split("/")[-1]
print("\nDownloading:", filename)
download(link, dst)
print("Downloaded:", filename)
# unzip zip file
print("Unzipping:", filename)
zip_path = os.path.join(dst, filename)
unzip(zip_path, dst)
print("Unzipped:", filename.replace(".zip", ""))
# remove zip file
os.remove(zip_path)
def download_dataset(download_dir: str):
"""
This script downloads the MIDV-500 dataset and unzips the folders.
"""
for link in all_links:
print('--------------------------------------------------------------')
# doownlaod zip file
print('\nDownloading:', link[40:])
download(link, download_dir)
print('Downloaded:', link[40:])
# unzip zip file
print('Unzipping:', link[40:])
zip_path = os.path.join(download_dir, link[40:])
unzip(zip_path, download_dir)
print('Unzipped:', link[40:].replace('.zip', ''))
# remove zip file
os.remove(zip_path)