def test_other_models(): for k in list_supported_models(): if k in ( 'sofvsr', 'vespcn', 'frvsr', 'qprn', 'ufvsr', 'yovsr', 'tecogan', 'spmc', 'rbpn' ): # skip video model continue train(k) eval(k)
# Update: 2020 - 2 - 10 import argparse from pathlib import Path from VSR.Backend import BACKEND from VSR.DataLoader import Dataset, Loader, load_datasets from VSR.Model import get_model, list_supported_models from VSR.Util import (Config, compat_param, save_inference_images, suppress_opt_by_args) parser = argparse.ArgumentParser( description=f'VSR ({BACKEND}) Testing Tool v1.0') g0 = parser.add_argument_group("basic options") g0.add_argument("model", choices=list_supported_models(), help="specify the model name") g0.add_argument("-p", "--parameter", help="specify the model parameter file (*.yaml)") g0.add_argument("-t", "--test", nargs='*', help="specify test dataset name or data path") g0.add_argument("--save_dir", default='../Results', help="working directory") g0.add_argument("--data_config", default="../Data/datasets.yaml", help="specify dataset config file") g1 = parser.add_argument_group("evaluating options") g1.add_argument( "--pretrain",
# Email: [email protected] # Update: 2020 - 2 - 10 import argparse from pathlib import Path from VSR.Backend import BACKEND from VSR.DataLoader import Dataset, Loader, load_datasets from VSR.Model import get_model, list_supported_models from VSR.Util import ( Config, compat_param, save_inference_images, suppress_opt_by_args ) parser = argparse.ArgumentParser(description=f'VSR ({BACKEND}) Testing Tool v1.0') g0 = parser.add_argument_group("basic options") g0.add_argument("model", choices=list_supported_models(), help="specify the model name") g0.add_argument("-p", "--parameter", help="specify the model parameter file (*.yaml)") g0.add_argument("-t", "--test", nargs='*', help="specify test dataset name or data path") g0.add_argument("--save_dir", default='../Results', help="working directory") g0.add_argument("--data_config", default="../Data/datasets.yaml", help="specify dataset config file") g1 = parser.add_argument_group("evaluating options") g1.add_argument("--pretrain", help="specify the pre-trained model checkpoint or will search into `save_dir` if not specified") g1.add_argument("--ensemble", action="store_true") g1.add_argument("--video", action="store_true", help="notify load test data as video stream") g2 = parser.add_argument_group("device options") g2.add_argument("--cuda", action="store_true", help="using cuda gpu") g2.add_argument("--threads", type=int, default=8, help="specify loading threads number") g3 = parser.add_argument_group("advanced options") g3.add_argument("--output_index", default='-1', help="specify access index of output array (slicable)") g3.add_argument("--export", help="export ONNX (torch backend) or protobuf (tf backend) (needs support from model)") g3.add_argument("--overwrite", action="store_true", help="overwrite the existing predicted output files")