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