欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

踩坑记录(opencv)

程序员文章站 2022-05-16 11:20:44
...

好久没更新了,记录一下项目中遇到的坑,希望大家看到会有帮助,同时也感谢那些给与我帮助的博客。
任务:给出一个视频,需要把视频顺时针旋转90度,再生成视频。(本次任务原视频宽大于高)
人生苦短,我用python
环境:Window10 Anaconda3 python3.7 opencv3.4.6 pycharm
代码如下:`

import cv2
import numpy as np

cap = cv2.VideoCapture("E:/python/test/video/1119_2.mov")

fps = int(cap.get(cv2.CAP_PROP_FPS))
print(fps)

fourcc = cv2.VideoWriter_fourcc(*'MJPG')
size=(int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)),int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)))   #旋转 经验3
print(size)     #(1080,1920)    (w,h)

videoWriter = cv2.VideoWriter('E:/python/test/video/1119_2_new.avi', fourcc, fps, size)

is_success, frame = cap.read()
print(is_success)
print(frame.shape)     #(h,w,c)

(h, w) = frame.shape[:2]
print(h,w)

center = (w // 2, w // 2)
padding = (w - h) // 2
M = cv2.getRotationMatrix2D(center, -90, 1.0)

print(M.shape)
#cv2.namedWindow("img", 0)

img_padded = np.zeros(shape=(w, w, 3), dtype=np.uint8)    #经验2
img_padded[padding:padding+h, :, :] = frame

while is_success:
   
    frame = cv2.warpAffine(img_padded, M, (w, w))  # 旋转90度

    frame = frame[:, padding:padding + h, :]
    print(frame.shape)

    # cv2.imshow("img",frame)
    # cv2.waitKey(5)   #  经验1  

    videoWriter.write(frame)
    is_success, frame = cap.read()
    if is_success:
        img_padded = np.zeros(shape=(w, w, 3), dtype=np.uint8)
        img_padded[padding:padding + h, :, :] = frame

videoWriter.release()
cap.release()
#cv2.destroyAllWindows()
`

经验1处表明,在循坏里面imshow的话,需要延时,可能由于处理时间很短。
经验2处表明,将图片拓展成正方形再旋转会有意向不到的好处。
经验3处表明,应该用代码打出视频的size,而不是看windows提供的属性方法,有时候会隔一点点。
写入VideoWriter对象的图片的尺寸要和定义VideoWriter对象的设置的size需要保持一致。
本次生成视频所设置的编码格式有效,只是生成的视频比较大,另外windows和Ubuntu都能打开生成的视频。