解決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()

 

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