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

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

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

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

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

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

Logo

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

更多推荐