diff --git a/src/detector.py b/src/detector.py index 2c6f351..2020fb5 100644 --- a/src/detector.py +++ b/src/detector.py @@ -76,7 +76,7 @@ class Detector: detections.append(Detection(label, confidences[i], box)) font = cv.FONT_HERSHEY_PLAIN - marked = cv.imread(filename) + marked = cv.imread(str(filename)) for detection in detections: x, y, w, h = detection.box cv.rectangle(marked, (x,y), (x + w, y + h), (255,255,255,0), 2)