diff --git a/yolo_opencv.py b/yolo_opencv.py index 5932e24..02a11d4 100644 --- a/yolo_opencv.py +++ b/yolo_opencv.py @@ -102,3 +102,4 @@ def draw_prediction(img, class_id, confidence, x, y, x_plus_w, y_plus_h): cv2.imwrite("object-detection.jpg", image) cv2.destroyAllWindows() +print('Done')