From 5bd2f9dcdedec74b0e25500bb11b1c2ad592b580 Mon Sep 17 00:00:00 2001
From: jiangxt <1579525634@qq.com>
Date: Tue, 8 Aug 2023 11:59:33 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E6=8C=87=E5=AE=9A=E5=B8=A7?=
 =?UTF-8?q?=E6=A0=87=E6=B3=A8=E4=BF=A1=E6=81=AF=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tool/video_label.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 tool/video_label.py

diff --git a/tool/video_label.py b/tool/video_label.py
new file mode 100644
index 0000000..98fa55a
--- /dev/null
+++ b/tool/video_label.py
@@ -0,0 +1,33 @@
+import cv2
+def video_label(video_path,labels,start_frame,end_frame):
+    """
+        给指定帧的图像，添加标注信息
+        video_path:视频输入文件
+        labels：动作标签
+        start_frame:动作起始帧
+        end_frame：动作结束帧
+    """
+    cap = cv2.VideoCapture(video_path)
+
+    while cap.isOpened():
+        ret, frame = cap.read()
+        if not ret:
+            break
+        frame_idx = cap.get(cv2.CAP_PROP_POS_FRAMES)
+        if start_frame <= frame_idx and frame_idx <= end_frame:
+            cv2.putText(frame, labels, (0, 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
+
+        cv2.namedWindow("Detection_results", cv2.WINDOW_AUTOSIZE)
+        cv2.imshow("Detection_results", frame)
+        if cv2.waitKey(10) & 0xFF == ord('q'):
+            break
+
+    cap.release()
+    cv2.destroyAllWindows()
+
+if __name__ == "__main__":
+    video_path = "D:/download/PaddleVideo1/output/output/after_1/0711-1_0_0.avi"
+    label = "playing phone!"
+    start_frame = 100
+    end_frame = 600
+    video_label(video_path,label,start_frame,end_frame)