mirror of https://github.com/YifanXu74/MQ-Det.git
27 lines
1.5 KiB
Python
27 lines
1.5 KiB
Python
import argparse
|
|
import os
|
|
|
|
argparser = argparse.ArgumentParser()
|
|
argparser.add_argument("--dataset_names", default="all", type=str) # "all" or names joined by comma
|
|
argparser.add_argument("--dataset_path", default="DATASET/odinw", type=str)
|
|
args = argparser.parse_args()
|
|
|
|
root = "https://vlpdatasets.blob.core.windows.net/odinw/odinw/odinw_35"
|
|
|
|
all_datasets = ["AerialMaritimeDrone", "AmericanSignLanguageLetters", "Aquarium", "BCCD", "ChessPieces", "CottontailRabbits", "DroneControl", "EgoHands", "HardHatWorkers", "MaskWearing", "MountainDewCommercial", "NorthAmericaMushrooms", "OxfordPets", "PKLot", "Packages", "PascalVOC", "Raccoon", "ShellfishOpenImages", "ThermalCheetah", "UnoCards", "VehiclesOpenImages", "WildfireSmoke", "boggleBoards", "brackishUnderwater", "dice", "openPoetryVision", "pistols", "plantdoc", "pothole", "selfdrivingCar", "thermalDogsAndPeople", "vector", "websiteScreenshots"]
|
|
|
|
datasets_to_download = []
|
|
if args.dataset_names == "all":
|
|
datasets_to_download = all_datasets
|
|
else:
|
|
datasets_to_download = args.dataset_names.split(",")
|
|
|
|
for dataset in datasets_to_download:
|
|
if dataset in all_datasets:
|
|
print("Downloading dataset: ", dataset)
|
|
os.system("wget " + root + "/" + dataset + ".zip" + " -O " + args.dataset_path + "/" + dataset + ".zip")
|
|
os.system("unzip " + args.dataset_path + "/" + dataset + ".zip -d " + args.dataset_path)
|
|
os.system("rm " + args.dataset_path + "/" + dataset + ".zip")
|
|
else:
|
|
print("Dataset not found: ", dataset)
|