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")