with a glass of water

This commit is contained in:
didier committed 2024-11-16 20:35:27 -05:00
1 parent a220035126
commit 31b9e85912
6 files changed
+115 -9

No files matched your search

+1 -3
View File
@@ -1,4 +1,3 @@
import json
from pathlib import Path
import logging
import torch
@@ -70,7 +69,6 @@ class DodgeballTracker:
self._data_dumper = data_dumper
self.detected_box_objects = sorted(detected_box_objects, reverse=True)
self.transformation_matrix = transformation_matrix
self.points = []
@pa.check_output(DetectedObject)
@@ -94,7 +92,7 @@ class DodgeballTracker:
load the pre-trained YOLOv8n model
"""
my_model = Path(
'/home/didier/all-dressed-programming/dodgeball-tracker/dodgeballtracker/runs/detect/train31/weights/best.pt')
'/home/didier/all-dressed-programming/dodgeball-tracker/runs/detect/train31/weights/best.pt')
return YOLO(my_model, verbose=False)
@pa.check_io(detected_objs=DetectedObject, out=TrackedDetectedObject)