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

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

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

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

上一篇:#力扣 LeetCode176. 第二高的薪水 @FDDLC


下一篇:leetcode176 第二高的薪水 Second Highest Salary