def get_original_model_info(self, original_model):
return ModelInfo(
input_space=original_model.input_space,
input_size=original_model.input_size,
input_range=original_model.input_range,
mean=original_model.mean,
std=original_model.std,
)
def get_original_model_info(self, original_model):
return ModelInfo(
input_space='RGB',
input_size=[3, 299, 299],
input_range=[0, 1],
mean=[0.5, 0.5, 0.5],
std=[0.5, 0.5, 0.5],
)
def get_original_model_info(self, original_model):
return ModelInfo(
input_space='RGB',
input_size=[3, 224, 224],
input_range=[0, 1],
mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225],
)
def get_original_model_info(self, original_model):
model_info = pretrainedmodels.pretrained_settings[self.model_name]
model_info = model_info[sorted(model_info.keys())[0]]
return ModelInfo(
input_space=model_info['input_space'],
input_size=model_info['input_size'],
input_range=model_info['input_range'],
mean=model_info['mean'],
std=model_info['std'],
)