File size: 885 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 |
from .test_options import TestOptions
class DetectOptions(TestOptions):
"""
继承TestOptions
新增推理图片路径、推理风格参数
"""
def initialize(self, parser):
parser = TestOptions.initialize(self, parser)
parser.set_defaults(name="detect")
parser.set_defaults(dataroot="imgs/", help="folder: /imgs/ or file: xx.jpg")
parser.set_defaults(checkpoints_dir="./weights")
parser.set_defaults(gpu_ids="-1")
# 新增
parser.add_argument("--no_save_img", action="store_true", help="no save fake_img")
parser.add_argument("--style", type=str, default="horse2zebra")
#
# parser.add_argument("--reload", action="store_true")
# parser.add_argument("--port")
# parser.add_argument("--reload-dir")
# parser.add_argument("--log-level")
return parser
|