From 3d5ffa6bdbd8dc8af49e09d2cd3c7975da351689 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Diego=20In=C3=A1cio?= Date: Wed, 12 Feb 2020 15:12:27 -0300 Subject: [PATCH] Update module object object_detection Change model path --- object_detection.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/object_detection.py b/object_detection.py index 3867d43..6eeb5c5 100644 --- a/object_detection.py +++ b/object_detection.py @@ -6,12 +6,12 @@ import numpy as np class ObjectDetection: def __init__(self): self.MODEL = cv2.dnn.readNet( - 'model/yolov3.weights', - 'model/yolov3.cfg' + 'models/yolov3.weights', + 'models/yolov3.cfg' ) self.CLASSES = [] - with open("model/coco.names", "r") as f: + with open("models/coco.names", "r") as f: self.CLASSES = [line.strip() for line in f.readlines()] self.OUTPUT_LAYERS = [self.MODEL.getLayerNames()[i[0] - 1] for i in self.MODEL.getUnconnectedOutLayers()]