YOLOv6 with SAHI

[Xavierhoyx]

Hi all,

I have trouble using SAHI with YOLOv6, when I try to define my detection model, I keep getting KeyError.

detection_model = AutoDetectionModel.from_pretrained(model_type='yolo6',model_path='weights/best_ckpt.pt',confidence_threshold=0.3,device="cuda:0")

I have used this line of code, replacing 'yolo6' with yolov7hub or yolov8 without any problems for SAHI with YOLOv7 & v8 respectively.

Any help would be appreciated, thank you.