深度神经网络对实时视频进行错误预测

原文标题deep neural network doing wrong predictions on Realtime videos

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

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

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

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

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

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

数据集链接

代码链接

原文链接:https://stackoverflow.com//questions/71894761/deep-neural-network-doing-wrong-predictions-on-realtime-videos

回复

我来回复
  • finko的头像
    finko 评论

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

    2年前 0条评论
  • dogekali的头像
    dogekali 评论

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

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

    2年前 0条评论