解决colab 无法imshow显示图像和视频

最近在做视频分析,电脑太烂,想用免费colab跑,但是colab上确无法像jupyter notebook那样正常使用cv2.imshow(),找了一些经实验有效的解决方案

#显示图片法一:用plt

import matplotlib.pyplot as plt

image = cv2.imread(path)

plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))

plt.show()

 

#显示图片法二 

from google.colab.patches import cv2_imshow

image = cv2.imread(path)

cv2_imshow(image)

 

#显示视频  目前还没有找到更好的方法,而是先一帧帧处理后合成视频存到google云盘上再用云盘预览,求大神指导

#colab 保存视频3部曲

import cv2

from google.colab.patches import cv2_imshow

#%matplotlib inline

 

video_path = "/content/drive/My Drive/dlib_test/test.avi"

video_path_put = "/content/drive/My Drive/dlib_test/test_out12345.avi"

cap = cv2.VideoCapture("/content/drive/My Drive/dlib_test/test.avi")

 

#step1 定义writer

fourcc = cv2.VideoWriter_fourcc(*"XVID")

writer = cv2.VideoWriter(video_path_put, fourcc, 30, (720, 576), True)  

                  #(image.shape[1], image.shape[0])

 

while cap.isOpened():

  grabbed, image = cap.read()

  if not grabbed:

    print ("Not grabbed.")

    break;

  if writer is None:

    # Initialize our video writer

    fourcc = cv2.VideoWriter_fourcc(*"XVID")

    writer = cv2.VideoWriter(video_path_put, fourcc, 30,

      (image.shape[1], image.shape[0]), True)

  #step2 写

  writer.write(image)

#step3 释放

writer.release()

cv2.destroyAllWindows()

cap.release()

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章