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