Cách xử lý hình ảnh của video, khung theo khung hình trong phát video bằng cách sử dụng python Opencv

Question

Tôi là người mới bắt đầu trong OpenCV. Tôi muốn thực hiện một số xử lý hình ảnh trên các khung hình của video đang được tải lên máy chủ của tôi. Tôi chỉ muốn đọc các khung hình có sẵn và viết chúng trong một thư mục. Và chờ một phần khác của video được tải lên và ghi các khung hình vào thư mục. Và tôi nên chờ cho mỗi khung hình được tải lên hoàn toàn rồi ghi nó vào một tập tin.

Bạn có thể cho tôi biết cách tôi có thể làm điều đó với OpenCV (Python) không?

Chỉnh sửa 1: Tôi đã viết mã này để quay video từ tệp, trong khi dữ liệu mới đang được thêm vào cuối tệp. Nói cách khác, tệp out.mp4 không phải là video hoàn chỉnh và một chương trình khác đang viết khung mới lên đó. Những gì tôi sẽ làm là, chờ đợi cho các chương trình khác để viết khung hình mới sau đó đọc chúng và hiển thị chúng.

Đây là mã của tôi:

import cv2 
cap = cv2.VideoCapture("./out.mp4") 

while True: 
    if cap.grab(): 
     flag, frame = cap.retrieve() 
     if not flag: 
      continue 
     else: 
      cv2.imshow('video', frame) 
    if cv2.waitKey(10) == 27: 
     break 

Vì vậy, vấn đề là cap.grab() gọi! khi không có khung, nó sẽ trả lại False! Và nó sẽ không chụp khung nữa, ngay cả khi tôi chờ đợi một thời gian dài.

Answer 1

Sau khi đọc tài liệu về VideoCapture. Tôi đã tìm ra rằng bạn có thể cho biết VideoCapture, khung nào được xử lý vào lần tới chúng tôi gọi là VideoCapture.read() (hoặc VideoCapture.grab()).

Vấn đề là khi bạn muốn read() một khung chưa sẵn sàng, đối tượng VideoCapture bị kẹt trên khung đó và không bao giờ tiếp tục. Vì vậy, bạn phải ép buộc nó bắt đầu lại từ khung trước đó.

Đây là mã

import cv2 

cap = cv2.VideoCapture("./out.mp4") 
while not cap.isOpened(): 
    cap = cv2.VideoCapture("./out.mp4") 
    cv2.waitKey(1000) 
    print "Wait for the header" 

pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES) 
while True: 
    flag, frame = cap.read() 
    if flag: 
     # The frame is ready and already captured 
     cv2.imshow('video', frame) 
     pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES) 
     print str(pos_frame)+" frames" 
    else: 
     # The next frame is not ready, so we try to read it again 
     cap.set(cv2.cv.CV_CAP_PROP_POS_FRAMES, pos_frame-1) 
     print "frame is not ready" 
     # It is better to wait for a while for the next frame to be ready 
     cv2.waitKey(1000) 

    if cv2.waitKey(10) == 27: 
     break 
    if cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES) == cap.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT): 
     # If the number of captured frames is equal to the total number of frames, 
     # we stop 
     break 

Answer 2

Trong tài liệu của OpenCV, có example để nhận khung hình video theo khung. Nó được viết bằng C++ nhưng rất dễ dàng để chuyển ví dụ sang python - bạn có thể tìm kiếm từng tài liệu fumction để xem cách gọi chúng trong python.

#include "opencv2/opencv.hpp" 

using namespace cv; 

int main(int, char**) 
{ 
    VideoCapture cap(0); // open the default camera 
    if(!cap.isOpened()) // check if we succeeded 
     return -1; 

    Mat edges; 
    namedWindow("edges",1); 
    for(;;) 
    { 
     Mat frame; 
     cap >> frame; // get a new frame from camera 
     cvtColor(frame, edges, CV_BGR2GRAY); 
     GaussianBlur(edges, edges, Size(7,7), 1.5, 1.5); 
     Canny(edges, edges, 0, 30, 3); 
     imshow("edges", edges); 
     if(waitKey(30) >= 0) break; 
    } 
    // the camera will be deinitialized automatically in VideoCapture destructor 
    return 0; 
} 

Answer 3

Đây là cách tôi sẽ bắt đầu để giải quyết này:

1. Tạo một nhà văn video:

   import cv2.cv as cv 
   videowriter = cv.CreateVideoWriter(filename, fourcc, fps, frameSize) 
   

   Check here for valid parameters

2. Vòng để lấy [1] và viết các khung:

   cv.WriteFrame(videowriter, frame) 
   

   WriteFrame doc

[1] zenpoy đã chỉ theo hướng đúng. Bạn chỉ cần biết rằng bạn có thể truy xuất hình ảnh từ webcam hoặc tệp :-)

Hy vọng rằng tôi đã hiểu đúng yêu cầu.

Answer 4

sử dụng này -

import cv2 
cap = cv2.VideoCapture('path of video file') 
count = 0 
while cap.isOpened(): 
    ret,frame = cap.read() 
    cv2.imshow('window-name',frame) 
    cv2.imwrite("frame%d.jpg" % count, frame) 
    count = count + 1 
    if cv2.waitKey(10) & 0xFF == ord('q'): 
     break 


cap.release() 
cap.destroyAllWindows() 

Answer 5

Theo các bản cập nhật mới nhất cho OpenCV 3.0 trở lên, bạn cần phải thay đổi định danh tài sản như sau trong mã bởi Mehran:

cv2.cv.CV_CAP_PROP_POS_FRAMES 

to

cv2.CAP_PROP_POS_FRAMES 

và áp dụng tương tự cho cv2.CAP_PROP_POS_FRAME_COUNT.

Hy vọng điều đó sẽ hữu ích.

Answer 6

Giải pháp duy nhất tôi tìm thấy là không đặt chỉ mục thành khung trước và chờ (sau đó OpenCV ngừng đọc khung hình), nhưng để khởi tạo lại lần chụp. Vì vậy, có vẻ như sau:

cap = cv2.VideoCapture(camera_url) 
while True: 
    ret, frame = cap.read() 

    if not ret: 
     cap = cv.VideoCapture(camera_url) 
     continue 

    # do your processing here 

Và nó hoạt động hoàn hảo!