search

gab-es21 / sea-turtles-detection

Detect sea turtles from drone imagery.
MIT License
1 stars 0 forks source link

Try all models #84

Open gab-es21 opened 11 months ago

gab-es21 commented 11 months ago
Model Filenames Task Inference Validation Training Export
YOLOv8 yolov8n.pt yolov8s.pt yolov8m.pt yolov8l.pt yolov8x.pt Detection ✅ ✅ ✅ ✅
Model Size (pixels) mAPval Speed CPU ONNX (ms) Speed A100 TensorRT (ms) Params (M) FLOPs (B)
YOLOv8n 640 37.3 80.4 0.99 3.2 8.7
YOLOv8s 640 44.9 128.4 1.20 11.2 28.6
YOLOv8m 640 50.2 234.7 1.83 25.9 78.9
YOLOv8l 640 52.9 375.2 2.39 43.7 165.2
YOLOv8x 640 53.9 479.1 3.53 68.2 257.8
gab-es21 commented 11 months ago

model = YOLO(f'{HOME}/yolov8n.pt') model = YOLO(f'{HOME}/yolov8s.pt') model = YOLO(f'{HOME}/yolov8m.pt') model = YOLO(f'{HOME}/yolov8l.pt') model = YOLO(f'{HOME}/yolov8xl.pt')