最近在做視頻分析,電腦太爛,想用免費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()