options/detect_options.py

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