用于图像有效性预测的AI

时间:2017-11-26 18:11:31

标签: tensorflow artificial-intelligence

我在数百台计算机上部署并运行全屏应用程序。有时应用程序崩溃并且屏幕显示Windows桌面或外部窗口弹出(系统,外部程序......)。我有几千个ok案例和数百个ko案例的截图,并希望将来自动化ko检测。 是否存在ai算法/ api,可以在输入中输入ok和ko截图并计算未来预测的模型?

1 个答案:

答案 0 :(得分:1)

听起来像卷积神经网络的标准工作。有很多教程可以引导您完成设置,并且有许多为您预先创建的良好网络。

我喜欢这个网站以获得良好的入门教程:

https://github.com/aymericdamien/TensorFlow-Examples

VGG,Inception,Mobilenet都是常见的卷积网络架构,在各种情况下表现都相当不错。

当你训练这样的模型时,最重要的是要知道平衡KO和OK类。如果您的KO示例少于OK示例,则会将算法偏向预测OK。您可以通过对KO示例进行过采样来解决此问题,只需根据需要重新使用它们,这样您就可以在两个类之间进行大约50/50的分割。

其他一切都应该很容易。获取像MNIST或CIFAR-10这样的CNN教程并进行设置。一旦您按照教程进行操作,就应该准备好在数据集上进行试用。

此外,您可能会缩小图像的大小,我认为在224x224范围内的某个位置通常可以在细节和计算时间之间取得良好的平衡。