def inference_and_crop(self, image_image_path, output_dir,
target_label_idx, **kwargs):
os.makedirs(output_dir, exist_ok=True)
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.model_config.as_dict())
driver.inference_and_crop(image_image_path, output_dir,
self.batch_size, target_label_idx, **kwargs)
def inference_and_extract(self, image_image_path, output_dir,
real_image_dir, label_dir, start_index,
random_seed, use_bbox_aug, bbox_aug_ratio,
**kwargs):
os.makedirs(output_dir, exist_ok=True)
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.model_config.as_dict())
driver.inference_and_extract(image_image_path, output_dir,
real_image_dir, label_dir, start_index,
self.batch_size, random_seed,
use_bbox_aug, bbox_aug_ratio, **kwargs)
def inference_single_image(self, image_image_path, output_dir, **kwargs):
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.image_size)
driver.inference(image_image_path, output_dir, **kwargs)
def inference_single_image(self, image_image_path, output_dir, **kwargs):
os.makedirs(output_dir, exist_ok=True)
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.model_config.as_dict())
driver.inference(image_image_path, output_dir, **kwargs)
def inference_single_image(self, image_image_path, output_dir, **kwargs):
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.image_size, self.num_classes,
self.enable_ema, self.data_format)
driver.inference(image_image_path, output_dir, **kwargs)
def inference_single_image(self, image_image_path, output_dir, **kwargs):
driver = inference.InferenceDriver(self.model_name, self.ckpt_path,
self.model_config.as_dict())
a = driver.inference(image_image_path, output_dir, **kwargs)
return a
