diff --git a/README.md b/README.md index f931a88..35eb090 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,7 @@ +# Right now writing detailed YOLO v3 tutorials for TensorFlow 2.x
+Detection and custom training process works better, is more accurate and has more planned features to do:
+https://github.com/pythonlessons/TensorFlow-2.x-YOLOv3 + # YOLOv3-object-detection-tutorial More information:
@@ -9,3 +13,9 @@ How to use YOLO v3 with webcam:
https://pylessons.com/YOLOv3-WebCam/

In future will update readme file + +Requirements:
+TensorFlow 1.15
+Keras 2.2.4
+Numpy (all versions should work)
+opencv-python (all versions should work)
diff --git a/YOLOv3-CSGO-detection/image_detect.py b/YOLOv3-CSGO-detection/image_detect.py index a809082..f3f9db9 100644 --- a/YOLOv3-CSGO-detection/image_detect.py +++ b/YOLOv3-CSGO-detection/image_detect.py @@ -277,6 +277,8 @@ def CreateXMLfile(image, ObjectsList): if len(ObjectsList) > 0: CreateXMLfile(cv2.imread(image), ObjectsList) cv2.imshow(image, r_image) + if cv2.waitKey(25) & 0xFF == ord("q"): + cv2.destroyAllWindows() cv2.imwrite('home_detection.jpg', r_image) yolo.close_session() diff --git a/YOLOv3-custom-training/image_detect.py b/YOLOv3-custom-training/image_detect.py index 621932c..b0dfd16 100644 --- a/YOLOv3-custom-training/image_detect.py +++ b/YOLOv3-custom-training/image_detect.py @@ -14,8 +14,7 @@ class YOLO(object): _defaults = { - #"model_path": 'logs/trained_weights_final.h5', - "model_path": 'logs/000/trained_weights_final.h5', + "model_path": 'logs/trained_weights_final.h5', "anchors_path": 'model_data/yolo_anchors.txt', "classes_path": '4_CLASS_test_classes.txt', "score" : 0.3, @@ -165,5 +164,6 @@ def detect_img(self, image): r_image, ObjectsList = yolo.detect_img(image) #print(ObjectsList) cv2.imshow(image, r_image) - + if cv2.waitKey(25) & 0xFF == ord("q"): + cv2.destroyAllWindows() yolo.close_session() diff --git a/YOLOv3-custom-training/realtime_detect.py b/YOLOv3-custom-training/realtime_detect.py index 38df5f9..7d036fc 100644 --- a/YOLOv3-custom-training/realtime_detect.py +++ b/YOLOv3-custom-training/realtime_detect.py @@ -197,6 +197,9 @@ def SHOWMSS_screen(p_output): r_image, ObjectsList = yolo.detect_image(img) cv2.imshow("YOLO v3", r_image) + if cv2.waitKey(25) & 0xFF == ord("q"): + cv2.destroyAllWindows() + return fps+=1 TIME = time.time() - start_time if (TIME) >= display_time :