with a glass of water
This commit is contained in:
1 parent
a220035126
commit
31b9e85912
6 files changed
+115
-9
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user