File size: 1,289 Bytes
58da73e |
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 |
from .base_options import BaseOptions
class TestOptions(BaseOptions):
"""
继承BaseOptions
"""
def initialize(self, parser):
parser = BaseOptions.initialize(self, parser) # define shared options
# 重写参数
self.isTrain = False
# To avoid cropping, the load_size should be the same as crop_size
parser.set_defaults(load_size=parser.get_default("crop_size"))
parser.set_defaults(model="test")
parser.set_defaults(no_dropout=True)
# 自定义部分
parser.set_defaults(name="horse2zebra")
parser.set_defaults(dataroot=r"datasets\horse2zebra\testA")
# 新增参数
parser.add_argument("--results_dir", type=str, default="./results/", help="saves results here.")
parser.add_argument("--aspect_ratio", type=float, default=1.0, help="aspect ratio of result images")
parser.add_argument("--phase", type=str, default="test", help="train, val, test, etc")
# Dropout and batch norm have different behaviors during training and test.
parser.add_argument("--eval", action="store_true", help="use eval mode during test time.")
parser.add_argument("--num_test", type=int, default=50, help="how many test images to run")
return parser
|