def testYolov3Detector(detectorargs):
mydetector = Yolov3Detector.MyYolov3Detector(detectorargs)
imgpath = os.path.join(
'/mnt/DATA5T/WaymoDataset/WaymoCOCO/Validation/validation_0000',
"11901761444769610243_556_000_576_000_1515475579357063_FRONT.jpg")
print(imgpath)
bbox_xyxy, pred_labels, cls_conf = detectimage.detectoneimage_novis(
imgpath, mydetector)
print(pred_labels)
def testDetectron2Detector(detectorargs):
mydetector = Detectron2Detector.MyDetectron2Detector(detectorargs)
imgpath=os.path.join('/mnt/DATA5T/WaymoDataset/WaymoCOCO/Validation/validation_0000', "11901761444769610243_556_000_576_000_1515475579357063_FRONT.jpg")
print(imgpath)
bbox_xyxy, pred_labels, cls_conf=detectimage.detectoneimage_novis(imgpath, mydetector)
print(pred_labels)
#Test a folder of image and output a video
outputvideopath='detectron2videoresult.mp4'
folderpath=os.path.join('/mnt/DATA5T/WaymoDataset/WaymoCOCO/Validation/', 'validation_0000')
detectimage.detectimagefolder_tovideo(folderpath, mydetector, outputvideopath)