深度神经网络对实时视频做出错误预测

deep neural network doing wrong predictions on Realtime videos

我使用 TensorFlow 创建了一个模型来检测视频中的任何类型的暴力行为。我已经训练了大约模型。 2000 个视频,将其分成帧。

但是当我在任何看不见的视频或实时视频上使用该模型时,它就无法正确预测。

我只是想问问是否有人可以告诉我我采用了正确的隐藏层,如果有任何调整我可以做出正确的预测。

neural_v2.ipynb用于训练模型。 test_v2.py 是加载模型并捕获视频和预测的文件。

如果您需要更多技术说明,请问我。

如果有人能以任何方式提供帮助,我将不胜感激。

Dataset Link

Code Link

你可以设置epochs=50再训练,会更好

理想情况下,您会将数据分成三部分:训练、验证和测试(您使用测试数据作为验证)。

作为@finko 的回答,我会尝试更多的时代,但更重要的是更密集的模型。试验一些最先进的模型(如 VGG16、ResNet152、MobileNet 等)。所有这些都可用作 Keras 应用程序 (https://www.tensorflow.org/api_docs/python/tf/keras/applications)。