File size: 577 Bytes
9d51df0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import argparse
# add command line arg for model type
parser = argparse.ArgumentParser()
parser.add_argument("--model_type", type=str, default="canny", help="Model type to download")
# add a binary flag to wipe the weights folder
parser.add_argument("--wipe", action="store_true", help="Wipe the weights folder")
args = parser.parse_args()
MODEL_TYPE = args.model_type
from utils import model_dl_urls, annotator_dl_urls, download_model
for model_name in annotator_dl_urls.keys():
download_model(model_name, annotator_dl_urls)
download_model(MODEL_TYPE, model_dl_urls)
|