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 :