基于机器学习自动识别螺栓并创建Bolt连接-2
上篇写了基于机器学习自动识别螺栓并创建Bolt Connector的内容。有很多感兴趣的朋友想更多的了解代码详情。我这边就不一一发邮件了。文后给出所有代码的网盘地址连接。这里在介绍一下需要注意的细节。
1.关于训练数据集的准备：本例中做的分类包含了三个部分，螺栓，螺母和其他部件。实际代码是可以做更多分类的。根据自己的需求做数据分类，并基于识别的结果做特定的处理，如钣金件，铸件的自动网格画分，连接件的自动连接，对称件特殊处理等等。
2.数据集的抓取：是在ANSA环境中自动截图完成的，输出的图片数据的风格，最好是保证实际使用机器学习模型时的风格和训练数据的风格保持一致。不然会影响预测精度。
3.ANSA环境所需的库：如果ANSA的使用环境是可以连接外网的，那不存在问题。如果是在公司内外使用，一般有两种方法：一个是找公司的IT支持部门，要到公司内网的python库源地址，然后安装库时引用该源地址即可。另一种就是把所需的库在外网环境下载好之后拷贝到内网环境下安装。如果既没有公司内网的python库源，又无法拷贝离线库文件。那就不要整这些了，公司不想让你乱搞。一般“正经”的公司都可以满足上面的条件。
4.关于提供的机器模型训练代码：不是我写的。我不会写，现在写代码不是都不需要自己写了嘛。会问就可以了。如上文所说，一般”正经”的公司，都有代码大模型可以用的。你没用可能是你不知道渠道。所以这里给的代码就是一个例子，具体到你的实际项目，还需要根据你的需求选择合适的模型训练你自己的模型。因为我给的代码，可能会由于我提问时的方式影响代码质量。
5.关于这个项目：只是一个小的例子，虽然可以落地。但有一定的使用门槛。对编程以及二次开发需要有一定的基础，如果只是针对螺栓等紧固件的自动建模，实际上ANSA本身是有这个功能的(这部分不需要机器学习包)。如果之前没有了解过的可以去详细了解一下，当然更多的内容也需要使用ML Toolkit这个工具包来完成，这个需要license。感兴趣的可以找官方了解下。（试用一般是三个月，通常可以多次试用--我用过，效果挺惊艳的）
6.关于代码分享：之前发了邮件的朋友收到的代码中，没有在ANSA中使用机器学习模型创建螺栓连接的部分。因为这部分只是涉及了使用ANSA API来创建螺栓连接的内容。这次一并补全，之前没有收到的朋友可以在文末的连接下载。但这里用到了两个函数，一个是创建螺栓connector，一个是realize connector。其实就是ansa的API。但由于里面有些不方便公开的部门，所有就给大家pyb格式的了。请见谅。但这个玩意儿有版本限制，因此我这里给了ANSA23(python3.8)、ANSA24(python3.11)和ANSA25(python3.11)版本的。其他版本的我不保证可用。
7.关于训练数据分享：我这里也没有理想的数据，展示的数据也都是网上随便下载的，其实用实际数据效果更好。所有我就不提供数据了，如果只是测试下代码，可以随便下载一些图片就可以，比如一些阿猫，阿狗的图片，训练一下就可以用来识别是狗还是猫了。至于什么牛马啊之类的，都可以的。
8.之前公众 号给我留言的朋友，没回复的。和大家说声见谅了。我平时看不到这些消息，用的那个"某星"手机也没有消息提醒。所有不是不回复大家，是超时无法回复了。见谅。我平时一般也不去后台看这些消息。
