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)