Does SAHI support yolov6? #971
Unanswered
Xavierhoyx
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
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 and yolov8 without any problems.
Any help would be appreciated, thank you.
Beta Was this translation helpful? Give feedback.
All reactions