二次开发与AI融合_基于tensorflow训练图像识别模型
书接上回,上一篇文章我们演示了如何基于预训练集识别图像,并且演示了螺栓的识别,由于训练集的模型也是现实世界的的图片,能准确的识别出现实世界中的螺栓。
但是,但是.....
当把CAD格式的螺栓图片输给AI,识别出来的结果奇奇怪怪,比如将螺母识别为培养皿🧫,虽然真的很相似,但是这个螺栓识别结果不具有参考性。
后台也有个小伙伴私信了,如何识别CAD的螺栓模型。
如果我们需要基于CAD的螺栓来识别模型,应该怎么处理呢。
解决办法也很简单,就是我们训练一个识别CAD的螺栓模型就可以了。第一步我们需要准备训练集,将训练集按照螺栓以及非螺栓两种状态 使用ImageDataGenerator来加载和预处理图像数据,将数据分为测试数据与训练数据,其中train_datagen用于训练和验证数据,也就是我们的学习的图片,test_datagen用于测试数据,就是我们验证的图片。 使用Sequential模型,包含卷积层、池化层、全连接层。再使用binary_crossentropy作为损失函数,adam作为优化器,构建模型。 使用fit方法训练模型,数据包含训练和验证数据,训练了10个epochs。 使用evaluate方法在测试数据上,评估模型性能。对训练模型的Python源码感兴趣的小伙伴可以选择付费,付费后私信,发送购买的截图,获取Hyper Mesh截图工具。而后可以自己对代码进行扩展。除了识别螺栓,可以识别你想识别的任何数据,只要您能提供训练数据。