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)