Spaces:
Runtime error
Runtime error
File size: 4,605 Bytes
7fab858 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
from .base_options import BaseOptions
class TestOptions(BaseOptions):
def initialize(self):
BaseOptions.initialize(self)
self.parser.add_argument("--ntest", type=int, default=float("inf"), help="# of test examples.")
self.parser.add_argument("--results_dir", type=str, default="./results/", help="saves results here.")
self.parser.add_argument(
"--aspect_ratio", type=float, default=1.0, help="aspect ratio of result images"
)
self.parser.add_argument("--phase", type=str, default="test", help="train, val, test, etc")
self.parser.add_argument(
"--which_epoch",
type=str,
default="latest",
help="which epoch to load? set to latest to use latest cached model",
)
self.parser.add_argument("--how_many", type=int, default=50, help="how many test images to run")
self.parser.add_argument(
"--cluster_path",
type=str,
default="features_clustered_010.npy",
help="the path for clustered results of encoded features",
)
self.parser.add_argument(
"--use_encoded_image",
action="store_true",
help="if specified, encode the real image to get the feature map",
)
self.parser.add_argument("--export_onnx", type=str, help="export ONNX model to a given file")
self.parser.add_argument("--engine", type=str, help="run serialized TRT engine")
self.parser.add_argument("--onnx", type=str, help="run ONNX model via TRT")
self.parser.add_argument(
"--start_epoch",
type=int,
default=-1,
help="write the start_epoch of iter.txt into this parameter",
)
self.parser.add_argument("--test_dataset", type=str, default="Real_RGB_old.bigfile")
self.parser.add_argument(
"--no_degradation",
action="store_true",
help="when train the mapping, enable this parameter --> no degradation will be added into clean image",
)
self.parser.add_argument(
"--no_load_VAE",
action="store_true",
help="when train the mapping, enable this parameter --> random initialize the encoder an decoder",
)
self.parser.add_argument(
"--use_v2_degradation",
action="store_true",
help="enable this parameter --> 4 kinds of degradations will be used to synthesize corruption",
)
self.parser.add_argument("--use_vae_which_epoch", type=str, default="latest")
self.isTrain = False
self.parser.add_argument("--generate_pair", action="store_true")
self.parser.add_argument("--multi_scale_test", type=float, default=0.5)
self.parser.add_argument("--multi_scale_threshold", type=float, default=0.5)
self.parser.add_argument(
"--mask_need_scale",
action="store_true",
help="enable this param meas that the pixel range of mask is 0-255",
)
self.parser.add_argument("--scale_num", type=int, default=1)
self.parser.add_argument(
"--save_feature_url", type=str, default="", help="While extracting the features, where to put"
)
self.parser.add_argument(
"--test_input", type=str, default="", help="A directory or a root of bigfile"
)
self.parser.add_argument("--test_mask", type=str, default="", help="A directory or a root of bigfile")
self.parser.add_argument("--test_gt", type=str, default="", help="A directory or a root of bigfile")
self.parser.add_argument(
"--scale_input", action="store_true", help="While testing, choose to scale the input firstly"
)
self.parser.add_argument(
"--save_feature_name", type=str, default="features.json", help="The name of saved features"
)
self.parser.add_argument(
"--test_rgb_old_wo_scratch", action="store_true", help="Same setting with origin test"
)
self.parser.add_argument("--test_mode", type=str, default="Crop", help="Scale|Full|Crop")
self.parser.add_argument("--Quality_restore", action="store_true", help="For RGB images")
self.parser.add_argument(
"--Scratch_and_Quality_restore", action="store_true", help="For scratched images"
)
self.parser.add_argument("--HR", action='store_true',help='Large input size with scratches')
|