Python中使用，YOLOv5实现视觉识别螺丝

异想实验室

4634人浏览 · 2022-01-16 16:15:54

异想实验室 · 2022-01-16 16:15:54 发布

实现过程中遍地是坑..各种报错...还好最终实现...

因为安装的某一些库的版本，不兼容的问题，导致无法使用GPU计算，所以视频中演示的效果，采用的是Cpu计算的，喂投数据集图像，41张，一共训练了200次，破本训练还是满慢的。。。

图像识别效果，特别注明，此图是存在数据集中的（主要是懒的拍新图了）..

下面视频中的，是实时摄像头画面识别。

黑色的是黑色的鼠标垫，并非刻意处理成黑色的效果。。

视觉识别Python中，使用YOLOv5实现识别螺丝效果

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

【开源项目】XIUNOX 基于 Xiuno BBS打造的现代化重构版本

全面适配 PHP 8 + MySQL 8，采用 Bootstrap 5.3 与 HTMX 4 构建现代无刷新 UI，安全与可扩展性大幅提升，原生支持多语言、RESTful API，让轻量论坛重获新生。：CSRF 防护 / XSS 防护（EscapeService）/ 参数化查询 / 登录限速 / 操作审计。htmx 4 驱动交互：局部刷新、乐观更新、morph DOM 保留，无刷新发帖 / 回复