コード例 #1
0
    def gather_options(self):
        if not self.initialized:
            parser = argparse.ArgumentParser(
                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
            parser = self.initialize(parser)

        opt, _ = parser.parse_known_args()

        model_option_setter = model.get_option_setter()
        parser = model_option_setter(parser, self.isTrain)
        opt, _ = parser.parse_known_args()

        self.parser = parser
        return parser.parse_args()
コード例 #2
0
    def gather_options(self):
        """Add additional model-specific options"""

        if not self.initialized:
            parser = self.initialize(self.parser)

        # get basic options
        opt, _ = parser.parse_known_args()

        # modify the options for different models
        model_option_set = model.get_option_setter(opt.model)
        parser = model_option_set(parser, self.isTrain)
        opt = parser.parse_args()

        return opt
コード例 #3
0
    def gather_options(self):
        """Add additional model-specific options"""

        if not self.initialized:
            self.initialize()
        # get basic options
        opt, _ = self.parser.parse_known_args()
        if opt.phase == 'train':
            self.isTrain = True
            self.add_train_options()
        else:
            self.isTrain = False
            self.add_test_options()
        # modify the options for different models
        model_option_set = model.get_option_setter(opt.model)
        parser = model_option_set(self.parser)
        opt = parser.parse_args()
        return opt
コード例 #4
0
    def gater_options(self):
        if not self.initialized:
            parser = argparse.ArgumentParser(
                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
            parser = self.initialize(parser)

        parser.parse_args(['--epoch_count', '2'])

        opt, _ = parser.parse_known_args()

        model_name = opt.model_name
        model_options_setter = model.get_option_setter(model_name)
        parser = model_options_setter(parser, self.isTrain)
        opt, _ = parser.parse_known_args()
        # assert(opt.data_name == opt.dataset_name.split('_')[1])

        self.parser = parser

        return parser.parse_args()
コード例 #5
0
ファイル: base_options.py プロジェクト: OkayMing/BPGAN
    def gather_options(self):
        if not self.initialized:  # check if it has been initialized
            parser = argparse.ArgumentParser(
                formatter_class=argparse.ArgumentDefaultsHelpFormatter)
            parser = self.initialize(parser)

        # get the basic options
        opt, _ = parser.parse_known_args()

        # modify model-related parser options
        # model_name = opt.model
        model_option_setter = model.get_option_setter()
        parser = model_option_setter(parser, self.isTrain)
        opt, _ = parser.parse_known_args()  # parse again with new defaults

        # modify dataset-related parser options
        # dataset_name = opt.dataset_mode
        dataset_option_setter = dataset.get_option_setter()
        parser = dataset_option_setter(parser, self.isTrain)

        # save and return the parser
        self.parser = parser
        return parser.parse_args()