def predict(self, imgPath, output):
predict_batch.mainDataset(imgPath, output,
'faster_rcnn_resnet50_v1b_custom',
self.pathPesos, self.classes)
def predict(self, imgPath, output):
predict_batch.mainDataset(imgPath, output, 'yolo3_darknet53_custom',
self.pathPesos, self.classes)
def predict(self, imgPath, output):
predict_batch.mainDataset(imgPath, output,
'ssd_512_resnet50_v1_custom', self.pathPesos,
self.classes)
def predict(self, dataset_path, cam_type, conf):
mainDataset(dataset_path, cam_type, conf, self)
