其他分享
首页 > 其他分享> > 使用pytorch搞定指针式仪表的读数

使用pytorch搞定指针式仪表的读数

作者:互联网

模型使用的是pytorch-cpn关键点检测网络
数据集自己使用labelme,打标图片为650张
打标效果如下(ps,有一些敏感信息打码了):
在这里插入图片描述
参考的作者的git地址如下:
https://github.com/GengDavid/pytorch-cpn
https://github.com/gathierry/FashionAI-KeyPointsDetectionOfApparel
经过数据集打标,然后整理成csv格式送到网络中调参训练,数据量比较小的时候,学习率要设置的低一点,训练时间耗时48h,最终效果如图:

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

标签:搞定,github,cpn,com,pytorch,https,指针式,打标
来源: https://blog.csdn.net/Enigma_tong/article/details/105261957