最近在做视频分析,电脑太烂,想用免费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()