首页/文章/ 详情

二次开发与AI融合_基于tensorflow训练图像识别模型

3月前浏览462


书接上回,上一篇文章我们演示了如何基于预训练集识别图像,并且演示了螺栓的识别,由于训练集的模型也是现实世界的的图片,能准确的识别出现实世界中的螺栓。

但是,但是.....

当把CAD格式的螺栓图片输给AI,识别出来的结果奇奇怪怪,比如将螺母识别为培养皿🧫,虽然真的很相似,但是这个螺栓识别结果不具有参考性。


后台也有个小伙伴私信了,如何识别CAD的螺栓模型。


如果我们需要基于CAD的螺栓来识别模型,应该怎么处理呢。

解决办法也很简单,就是我们训练一个识别CAD的螺栓模型就可以了。

第一步我们需要准备训练集,将训练集按照螺栓以及非螺栓两种状态

代码由以下这几部分组成
  • 数据生成器:
    使用ImageDataGenerator来加载和预处理图像数据,将数据分为测试数据与训练数据,其中train_datagen用于训练和验证数据,也就是我们的学习的图片,test_datagen用于测试数据,就是我们验证的图片。
  • 模型构建:
    使用Sequential模型,包含卷积层、池化层、全连接层。再使用binary_crossentropy作为损失函数,adam作为优化器,构建模型。
  • 模型训练:
    使用fit方法训练模型,数据包含训练和验证数据,训练了10个epochs
  • 模型评估:
    使用evaluate方法在测试数据上,评估模型性能。
  • 显示预测结果:
    和前文一样调用PIL显示预测的图片及其预测结果。

对训练模型的Python源码感兴趣的小伙伴可以选择付费,付费后私信,发送购买的截图,获取Hyper Mesh截图工具。
而后可以自己对代码进行扩展。除了识别螺栓,可以识别你想识别的任何数据,只要您能提供训练数据。
当然Python代码不管基于那个平台都可以的。

来源:TodayCAEer
二次开发python螺栓gEDA
著作权归作者所有,欢迎分享,未经许可,不得转载
首次发布时间:2024-08-14
最近编辑:3月前
TodayCAEer
本科 签名征集中
获赞 18粉丝 27文章 242课程 0
点赞
收藏
作者推荐

HyperMesh二次开发:将tcl格式字典转化为Python格式字典

在进行HyperMesh二次开发中,我们常常面临tcl语言的一些不完善之处。为了实现特定功能,我们需要调用Python来进行一些处理。其中,参数传递是一个重要的环节。当涉及到参数传递时,毫无疑问字典是最具可读性的数据结构。然而,由于tcl语言和Python在字典格式上存在差异,我们需要找到一种方法来实现格式转换。为了解决这个问题,可以使用Python来实现tcl格式字典到Python格式字典的转换。首先,我们将tcl格式的字典转换为Python字符串。通过匹配tcl字典关键字“{”或者“}”,来判断是否存在嵌套,如果存在则继续递归处理,主程序就根据tcl格式字符串的特定结构即一一对应的格式,我们可以解析出键值对。然后,我们将解析出的键值对添加到Python的字典中,得到最终的Python格式字典。通过调用这个函数,我们可以将tcl格式的字典转换为Python字典。程序调取方式importTclDictToPythonDicttcl_dict_str="key0{key0{key0{key0value0}}}key1{key0{key0{key0value0key1value1}key1{key0value0}key2{key0value0key1value1}}key1{key0{key0value0key1value1key2value2key3value3}key1{key0value0key1value1key2value2key3value3}key2{key0value0}}key2{key0{key0value0key1value1}key1{key0value0key1value1}key2{key0value0key1value1}}key3{key0{key0value0key1value1key2value2}key1{key0value0key1value1key2value2key3value3}}}key2{key0{key0{key0value0key1value1key2value2key3value3key4value4}}}key3{key0{key0{key0value0key1value1key2value2key3value3key4value4}key1{key0value0key1value1key2value2key3value3}}key1{key0{key0value0key1value1key2value2key3value3}key1{key0value0key1value1key2value2key3value3key4value4}key2{key0value0}}key2{key0{key0value0key1value1}}}"python_dict=TclDictToPythonDict.parse_tcl_dict(tcl_dict_str)print(python_dict)在HyperMesh二次开发中,将tcl格式字典转化为Python格式字典,不仅可以提高代码的可读性和灵活性,还可以结合两种语言的优势,更加高效地实现所需功能。程序获取方式,后台回复关键字,“字典转换”来源:TodayCAEer

未登录
还没有评论
课程
培训
服务
行家
VIP会员 学习 福利任务 兑换礼品
下载APP
联系我们
帮助与反馈