Sample of the YOLO v6 Daisi Code

yolo_v6_daisi_sample.py

def yolo(image: np.ndarray=None, return_type: list=["Image", "Labels"]):
    ...
    inferer = Inferer(source, weights, device="", yaml="coco.yaml", 
                      img_size=new_width, half=False)
    inferer.infer(conf_thres=.25, iou_thres=.45, classes=None, 
                  agnostic_nms=False, max_det=1000, save_dir=tmpdir, 
                  save_txt=save_txt, save_img=save_img, hide_labels=False, hide_conf=False)
    ...
    return yolo_result, label_result