def METHOD_NAME(self):
    if enable_gpu == "ON":
        self.target = DefaultNVGPUTarget()
    else:
        self.target = DefaultHostTarget()
    self.model_dir = model_dir
    self.x_shape = [1, 3, 224, 224]
    self.target_tensor = 'save_infer_model/scale_0.tmp_1'
    self.input_tensor = 'inputs'