• python opencv 读取mp4,上一帧，下一帧

    目录

    安装pynput：

    python opencv 读取mp4, 有上一帧，下一帧的功能

    队列 缓存最新版，有帧号

    队列封装成类版本：

    安装pynput：

    pip install pynput

    python opencv 读取mp4, 有上一帧，下一帧的功能

    这个每次读取self.cap.set(cv2.CAP_PROP_POS_FRAMES, frame_index)，后退读取稍微有点慢，

    1. import cv2
    2. from pynput import keyboard
    3.  
    4. class VideoPlayer:
    5.     def __init__(self, video_path):
    6.         self.cap = cv2.VideoCapture(video_path)
    7.         self.frame_index = -1
    8.         self.total_frames = int(self.cap.get(cv2.CAP_PROP_FRAME_COUNT))
    9.         self.new_frame_flag = False
    10.         self.last_frame_i=0
    11.  
    12.     def show_frame(self, frame_index):
    13.         if 0 <= frame_index < self.total_frames:
    14.             if frame_index
    15.                 self.cap.set(cv2.CAP_PROP_POS_FRAMES, frame_index)
    16.             self.last_frame_i = frame_index
    17.             ret, frame = self.cap.read()
    18.             if ret:
    19.                 cv2.imshow('Video', frame)
    20.                 self.frame_index = frame_index
    21.                 self.new_frame_flag = False
    22.  
    23.     def next_frame(self):
    24.         if self.frame_index + 1 < self.total_frames:
    25.             self.frame_index += 1
    26.             self.new_frame_flag = True
    27.  
    28.     def prev_frame(self):
    29.         if self.frame_index - 1 >= 0:
    30.             self.frame_index -= 1
    31.             self.new_frame_flag = True
    32.  
    33.     def on_press(self, key):
    34.         if key == keyboard.Key.right:  # Right arrow key
    35.             self.next_frame()
    36.         elif key == keyboard.Key.left:  # Left arrow key
    37.             self.prev_frame()
    38.         elif str(key) == "'q'":  # Press 'q' to exit
    39.             return False
    40.  
    41.     def play(self):
    42.         listener = keyboard.Listener(on_press=self.on_press)
    43.         listener.start()
    44.  
    45.         self.show_frame(0)  # Show the first frame
    46.  
    47.         while True:
    48.             if self.new_frame_flag:
    49.                 self.show_frame(self.frame_index)
    50.             if cv2.waitKey(1) & 0xFF == ord('q'):  # Press 'q' to exit
    51.                 break
    52.  
    53.         cv2.destroyAllWindows()
    54.  
    55. if __name__ == "__main__":
    56.     player = VideoPlayer(rf'F:\data\9.mp4')
    57.     player.play()

    队列 缓存最新版，有帧号

    1. import cv2
    2. from collections import deque
    3. from pynput import keyboard
    4.  
    5.  
    6. def on_press(key):
    7.     global queue_index, load_new_frame
    8.     if key == keyboard.Key.right or key == keyboard.Key.space:  # Right arrow key
    9.         if queue_index < len(frame_buffer) - 1:
    10.             queue_index += 1
    11.         else:
    12.             load_new_frame = True
    13.     elif key == keyboard.Key.left:  # Left arrow key
    14.         if queue_index > 0:
    15.             queue_index -= 1
    16.         load_new_frame = False
    17.     elif str(key) == "'q'":  # Press 'q' to exit
    18.         return False
    19.  
    20.  
    21. listener = keyboard.Listener(on_press=on_press)
    22. listener.start()
    23.  
    24. if __name__ == '__main__':
    25.     cap = cv2.VideoCapture(r'C:\Users\Administrator\Documents\youjie\zhuanchang\11.mp4')
    26.  
    27.     deque_len = 300
    28.     frame_buffer = deque(maxlen=deque_len)
    29.     queue_index = -1
    30.     load_new_frame = True
    31.     frame_index = -1
    32.     while cap.isOpened():
    33.         if load_new_frame:
    34.             ret, frame = cap.read()
    35.             if not ret:
    36.                 print('video end')
    37.                 continue
    38.  
    39.             frame_index += 1
    40.             cv2.putText(frame, str(frame_index), (20, 70), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 0, 255), 2)
    41.             frame_buffer.append(frame)
    42.             queue_index = len(frame_buffer) - 1
    43.             # print('load frame_i', frame_index, "queue_i", queue_index, 'queue len', len(frame_buffer))
    44.             load_new_frame = False
    45.  
    46.         # Show the current frame
    47.         if len(frame_buffer) > 0:
    48.             view_index = frame_index - len(frame_buffer) + queue_index + 1
    49.             imshow = frame_buffer[queue_index]
    50.             # print('show frame_i', frame_index, "queue_i", queue_index, 'view_i', view_index)
    51.             if max(imshow.shape[:2])>1280:
    52.                 imshow = cv2.resize(imshow, (1280, 720))
    53.             cv2.imshow('Video', imshow)
    54.  
    55.         if cv2.waitKey(1) & 0xFF == ord('q'):  # Press 'q' to exit
    56.             break
    57.  
    58.     # Release the video and close windows
    59.     cap.release()
    60.     cv2.destroyAllWindows()

    队列封装成类版本：

    1. import glob
    2. import os
    3. import re
    4.  
    5. import cv2
    6. from collections import deque
    7.  
    8. import yaml
    9. from pynput import keyboard
    10.  
    11. from a2_jupter_check.ip_result.track_tool import get_box_dict, get_color
    12.  
    13. def show_box(track_id, box, black_image):
    14.     color = get_color(box['sku'])
    15.     cv2.rectangle(black_image, (box['box'][0], box['box'][1]), (box['box'][2], box['box'][3]), color, 1)
    16.     cv2.putText(black_image, f"p:{(box['prob']):.3f}", (box['box'][0], box['box'][1] + 20), cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
    17.     cv2.putText(black_image, "t:" + str(track_id), (box['box'][0], box['box'][1] + 40), cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
    18.     cv2.putText(black_image, "s:" + str(box['sku']), (box['box'][0], box['box'][1] + 60), cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
    19.  
    20.  
    21. class VideoPlayer:
    22.     def __init__(self, video_path,video_path2,track_txt):
    23.  
    24.         self.cap = cv2.VideoCapture(video_path)
    25.         if door_type==1:
    26.             self.cap2 = cv2.VideoCapture(video_path2)
    27.         self.frame_index = -1
    28.         self.total_frames = int(self.cap.get(cv2.CAP_PROP_FRAME_COUNT))
    29.         if door_type == 1:
    30.             self.total_frames2 = int(self.cap2.get(cv2.CAP_PROP_FRAME_COUNT))
    31.             self.total_frames=min(self.total_frames,self.total_frames2)
    32.  
    33.         self.top_dict,  self.bot_dict,  self.end_index = get_box_dict(track_txt)
    34.         self.queue_index = -1
    35.     def save_cls_err(self,cam_id):
    36.         if cam_id==0:
    37.             box_id=0
    38.             for box in self.top_boxes:
    39.                 img_crop=self.top_image_o[box['box'][1]:box['box'][3],box['box'][0]:box['box'][2]]
    40.                 save_path=f"{cls_dir}/{box['sku']}_{(box['prob']):.2f}_{self.frame_index}_0_{box_id}.jpg"
    41.                 cv2.imwrite(save_path,img_crop)
    42.                 print('cls top save img',save_path)
    43.                 box_id+=1
    44.  
    45.         elif cam_id==1:
    46.             box_id = 0
    47.             for box in self.bot_boxes:
    48.                 img_crop = self.bot_image_o[box['box'][1]:box['box'][3], box['box'][0]:box['box'][2]]
    49.                 save_path=f"{cls_dir}/{box['sku']}_{(box['prob']):.2f}_{self.frame_index}_1_{box_id}.jpg"
    50.                 cv2.imwrite(save_path, img_crop)
    51.                 print('cls bot save img',save_path)
    52.                 box_id += 1
    53.     def on_press(self, key):
    54.         global box_offset
    55.         if key == keyboard.Key.right or key == keyboard.Key.space:  # Right arrow key
    56.             if self.queue_index < len(frame_buffer) - 1:
    57.                 self.queue_index += 1
    58.             else:
    59.                 self.load_new_frame = True
    60.         elif key == keyboard.Key.left:  # Left arrow key
    61.             if self.queue_index > 0:
    62.                 self.queue_index -= 1
    63.             self.load_new_frame = False
    64.         elif key == keyboard.KeyCode.from_char('+'):  # Numpad +
    65.             box_offset+=1
    66.             print("box_offset +1", box_offset)
    67.             self.new_frame_flag = True
    68.         elif key == keyboard.KeyCode.from_char('-'):  # Numpad -
    69.             box_offset-=1
    70.             print("box_offset -1", box_offset)
    71.             self.new_frame_flag = True
    72.  
    73.         elif hasattr(key, 'char') and key.char =='1':
    74.             self.save_cls_err(0)
    75.         elif hasattr(key, 'char') and key.char =='2':
    76.             self.save_cls_err(1)
    77.         elif hasattr(key, 'char') and key.char =='3':
    78.             save_path = f"{det_dir}/{sw_name}_{self.frame_index}_0.jpg"
    79.             cv2.imwrite(save_path, self.top_img_big)
    80.             print('save det err cam 0', save_path)
    81.         elif hasattr(key, 'char') and key.char =='4':
    82.             save_path = f"{det_dir}/{sw_name}_{self.frame_index}_1.jpg"
    83.             cv2.imwrite(save_path, self.bot_img_big)
    84.             print('save det err cam 1',save_path)
    85.         elif str(key) == "'c'":  # 's' key
    86.             print('c')
    87.         elif str(key) == "'d'":  # 's' key
    88.             print('d')
    89.         elif str(key) == "'q'":  # Press 'q' to exit
    90.             return False
    91.         else:
    92.             try:
    93.                 if key.char == '/':  # 检查按键字符
    94.                     print('Numpad / key pressed')
    95.                 elif key.char=='*':
    96.                     print('Numpad * key pressed')
    97.             except AttributeError:
    98.                 pass
    99.     def show_frame(self):
    100.         listener = keyboard.Listener(on_press=self.on_press)
    101.         listener.start()
    102.         self.load_new_frame = True
    103.         while self.cap.isOpened():
    104.             if self.load_new_frame:
    105.                 ret, frame = self.cap.read()
    106.                 if not ret:
    107.                     print('video end')
    108.                     continue
    109.  
    110.                 self.frame_index += 1
    111.                 cv2.putText(frame, str(self.frame_index), (20, 70), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 0, 255), 2)
    112.                 frame_buffer.append(frame)
    113.                 self.queue_index = len(frame_buffer) - 1
    114.                 # print('load frame_i', self.frame_index, "queue_i", queue_index, 'queue len', len(frame_buffer))
    115.                 self.load_new_frame = False
    116.  
    117.             # Show the current frame
    118.             if len(frame_buffer) > 0:
    119.                 view_index = self.frame_index - len(frame_buffer) + self.queue_index + 1
    120.                 imshow = frame_buffer[self.queue_index]
    121.                 # print('show frame_i', self.frame_index, "queue_i", queue_index, 'view_i', view_index)
    122.                 cv2.imshow('Video', imshow)
    123.  
    124.             if cv2.waitKey(1) & 0xFF == ord('q'):  # Press 'q' to exit
    125.                 break
    126.  
    127.         # Release the video and close windows
    128.         self.cap.release()
    129.         cv2.destroyAllWindows()
    130. if __name__ == '__main__':
    131.  
    132.     sbbh='2212311106000040'
    133.  
    134.     dir_a= r'F:\data\jupiter\err_nos\0912\569'
    135.     dir_a= r'D:\data\err_orders\jupter_pair_09\SW0031423091206302824557'
    136.     dir_a= r'D:\data\err_orders\jupter_pair_09\SW0031423091206323752502'
    137.     cooler_path = rf'F:\data\jupiter\configs/{sbbh}/config/config_cooler.yaml'
    138.  
    139.     with open(cooler_path, 'r') as file:
    140.         yaml_dict = yaml.safe_load(file)
    141.     img_x=yaml_dict['img_x']
    142.     # img_x = 1920 - 1240 - img_x
    143.     img_y=yaml_dict['img_y']
    144.     top_line=yaml_dict['top_line']
    145.     bot_line=yaml_dict['bot_line']
    146.     img_height, img_width = 744, 1240
    147.     door_type = 1
    148.  
    149.     cls_dir=dir_a+'/cls'
    150.     det_dir=dir_a+'/det'
    151.     os.makedirs(cls_dir,exist_ok=True)
    152.     os.makedirs(det_dir,exist_ok=True)
    153.     mp4_files = glob.glob(dir_a + '/high_*.mp4')
    154.     # img_x=300+75
    155.     # img_y= 202
    156.     box_offset= 0
    157.     video_path=video_path2=None
    158.     for file in mp4_files:
    159.         if "high_" in file and "_0_" in file:
    160.             video_path=file
    161.         elif "high_" in file and "_1_" in file:
    162.             video_path2=file
    163.         elif "high_" in file and "_2_" in file:
    164.             video_path=file
    165.             door_type = 2
    166.             print("door_type=2")
    167.  
    168.     if door_type==1 and (video_path is None or video_path2 is None):
    169.         print("cap1 or cap2 is None")
    170.         exit(0)
    171.     txt_path = video_path.replace(".mp4", "_tracks.txt")
    172.  
    173.     pattern = r'SW\d+'
    174.     match = re.search(pattern, txt_path)
    175.     sw_name = ""
    176.     if match:
    177.         sw_name = match.group()
    178.     if door_type==2:
    179.         img_x = 1920 - 1240 - img_x
    180.  
    181.     deque_len = 30
    182.     frame_buffer = deque(maxlen=deque_len)
    183.  
    184.  
    185.     save_width = 1000
    186.     save_height = save_width *(img_height * 2)  // img_width
    187.     mp4_path=os.path.dirname(txt_path)+'/'+sw_name+'_tracks.mp4'
    188.     fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 或者使用 'X264'
    189.     out_writer = cv2.VideoWriter(mp4_path, fourcc, 25.0, (save_width, save_height))
    190.     print('save track mp4 ',mp4_path,'w',save_width,'h',save_height)
    191.     player = VideoPlayer(video_path,video_path2, txt_path)
    192.     player.show_frame()

  • 相关阅读:
    如何更改代理ip，变更代理ip怎么实现？
    Allegro只给孔或pin加背钻属性操作指导
    Day19：属性封装详解
    xxl-job 数据库由mysql替换为postgre
    如何利用Python中实现高效的网络爬虫
    【freertos】002-posix模拟器设计与cortex m3异常处理
    年轻人应该做长期并能提升自己的兼职副业
    【ICLR 2021】半监督目标检测：Unbiased Teacher For Semi-Supervised Object Detection
    431. 将 N 叉树编码为二叉树 DFS
    Kubernetes k8s 平台部署常用命令
  • 原文地址：https://blog.csdn.net/jacke121/article/details/132843821