adding recorder yolov4.py

Author: Asadullah-Dal17

.gitignore

@@ -0,0 +1,2 @@
+*.mp4
+*.avi

yolov4.py

@@ -1,4 +1,5 @@
 import cv2 as cv
+import time
 Conf_threshold = 0.4
 NMS_threshold = 0.4
 COLORS = [(0, 255, 0), (0, 0, 255), (255, 0, 0),
@@ -17,9 +18,11 @@
 
 
 cap = cv.VideoCapture('output.avi')
-
+starting_time = time.time()
+frame_counter = 0
 while True:
     ret, frame = cap.read()
+    frame_counter += 1
     if ret == False:
         break
     classes, scores, boxes = model.detect(frame, Conf_threshold, NMS_threshold)
@@ -28,8 +31,12 @@
         label = "%s : %f" % (class_name[classid[0]], score)
         cv.rectangle(frame, box, color, 1)
         cv.putText(frame, label, (box[0], box[1]-10),
-                   cv.FONT_HERSHEY_COMPLEX, 0.5, color, 2)
-
+                   cv.FONT_HERSHEY_COMPLEX, 0.3, color, 1)
+    endingTime = time.time() - starting_time
+    fps = frame_counter/endingTime
+    # print(fps)
+    cv.putText(frame, f'FPS: {fps}', (20, 50),
+               cv.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), 2)
     cv.imshow('frame', frame)
     key = cv.waitKey(1)
     if key == ord('q'):

yolov4_Recording.py

@@ -0,0 +1,63 @@
+import cv2 as cv
+import time
+Conf_threshold = 0.4
+NMS_threshold = 0.4
+COLORS = [(0, 255, 0), (0, 0, 255), (255, 0, 0),
+          (255, 255, 0), (255, 0, 255), (0, 255, 255)]
+
+class_name = []
+with open('classes.txt', 'r') as f:
+    class_name = [cname.strip() for cname in f.readlines()]
+# print(class_name)
+net = cv.dnn.readNet('yolov4-tiny.weights', 'yolov4-tiny.cfg')
+net.setPreferableBackend(cv.dnn.DNN_BACKEND_CUDA)
+net.setPreferableTarget(cv.dnn.DNN_TARGET_CUDA_FP16)
+
+model = cv.dnn_DetectionModel(net)
+model.setInputParams(size=(416, 416), scale=1/255, swapRB=True)
+
+
+cap = cv.VideoCapture('new.mp4')
+frame_width = cap.get(cv.CAP_PROP_FRAME_WIDTH)
+frame_height = cap.get(cv.CAP_PROP_FRAME_HEIGHT)
+print(frame_height, frame_width)
+fourcc = cv.VideoWriter_fourcc('M', 'J', 'P', 'G')
+# cap.set(cv.CAP_PROP_FPS, 7)
+dim = (int(frame_width), int(frame_height))
+out = cv.VideoWriter('OutputVideo2.avi', fourcc, 30.0, dim)
+starting_time = time.time()
+frame_counter = 0
+while True:
+    ret, frame = cap.read()
+
+    frame_counter += 1
+    if ret == False:
+        break
+
+    # if frame_counter == 100:
+        # break
+
+    frame = cv.resize(frame, dim, interpolation=cv.INTER_AREA)
+    classes, scores, boxes = model.detect(frame, Conf_threshold, NMS_threshold)
+    for (classid, score, box) in zip(classes, scores, boxes):
+        color = COLORS[int(classid) % len(COLORS)]
+        label = "%s : %f" % (class_name[classid[0]], score)
+        cv.rectangle(frame, box, color, 4)
+        cv.putText(frame, label, (box[0], box[1]-10),
+                   cv.FONT_HERSHEY_COMPLEX, 0.6, color, 4)
+    endingTime = time.time() - starting_time
+    fps = frame_counter/endingTime
+    # print(fps)
+    cv.putText(frame, f'FPS: {round(fps,2)}', (20, 50),
+               cv.FONT_HERSHEY_COMPLEX, 0.7, (0, 255, 0), 2)
+    # cv.imshow('frame', frame)
+
+    out.write(frame)
+    key = cv.waitKey(1)
+    if key == ord('q'):
+        break
+out.release()
+
+cap.release()
+cv.destroyAllWindows()
+print('done')