fashionAI 服装关键点检测
,给定五种类型的服装,采用人体姿态估计的方法检测关键点。最终结果排名24,CPN模型没有复现很好,略遗憾...
0.效果预览
1.模型
模型结合了HourGlass模型和CPN模型,其中HG堆叠了2个,另外在HG上采样过程的1/2大小的特征图上添加了热点图监督标签。
RGB图像送入两个分支网络分别计算,最后concat二者的特征图,具体结构如图所示。
添加了soft-argmax层,可以由热点图转化到具体的坐标值,用坐标值groundtruth监督学习
2.策略
最多只能使用两个不同参数的模型,检测模型也算。通过检测可以提高目标占比,提升效果。
使用第一级预测结果截取目标,为了防止截取不完整,向外扩展30像素,再训练第二级crop模型。
第一级模型testB线上4.17%,crop之后的模型testB线上4.05%,融合之后3.95%.
3.训练与预测细
1