cv2_imshow collab不显示任何用于使用cnn+l预测视频的内容;stm模型

Question

嗨，伙计们，我正在尝试将我的模型测试为预测的视频

我没有错误日志或任何

但什么都没有出现，我正试图打开视频，并使用我的cnn(alexnet(+lstm模型预测它

这是我的代码

model_path = 'Model/CCTV_10Frame_Adam_Model_1e2_b16_regul.h5'
model = models.load_model(model_path)
vid = cv2.VideoCapture("Data16_116.mp4")
while(True):
ret, frame = vid.read()
start = time.time()
if ret:
draw = frame.copy()
draw = cv2.cvtColor(draw, cv2.COLOR_BGR2RGB)
scale_percent = 20 # percent of original size
width = 299
height = width
dim = (width, height)
frame_set = cv2.resize(draw, dim, interpolation = cv2.INTER_AREA)
frame_set=np.arange(10*width*height*3).reshape(10,width, height, 3)
frame_set.reshape(10, width, height, 3).shape
frame_set = np.expand_dims(frame_set, axis=0)
result=model.predict(frame_set)

cv2_imshow(result)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
break
vid.release()
cv2.destroyAllWindows()

有人知道这件事吗？

任何答案都将感谢

非常感谢！

注意：我确实在vsc中尝试过，但遗憾的是，它只打开了1秒，日志显示

分配器(GPU_0_bfc(在尝试分配1.25GiB时内存不足其中freed_by_count＝0。调用者表示这不是失败，但可能意味着如果更多内存可用。

Answer 1

"quot"你能在从代码中删除**else:break**块后进行检查吗？因为它可能会跳过相机帧而不验证用户的任何中断。"quot">

if cv2.waitKey(1) & 0xFF == ord('q'): // remove the else block
break