项目导读
服饰分割的主要场景是对模特进行服饰抠图,要求边缘处理相对平滑,扣取召回和准确率比较高,能够覆盖95%以上的场景case。同时需要考虑模型FLOPs以及结构便宜性,便于后期有压缩的需求。本文记录了一个服饰分割项目的全流程,希望能给大家带来一些帮助。
一、序
很久没有写过博客了,最近忙于做项目,闭了后简单写一下心得体会。近期主要是在做服饰场景相关的项目,今天简单写写其中做的一个服饰分割。
二、背景
主要场景就是对模特进行服饰抠图,要求边缘处理相对平滑,扣取召回和准确率比较高,能够覆盖95%以上的场景case。同时需要考虑模型FLOPs以及结构便宜性,便于后期有压缩的需求。
三、模型选择
有考虑过如下三种模型:
- DeepLabV3
- U2Net
- HRNet-seg
1.这里DeepLabV3有空洞卷积存在,对于细致的扣图,效果不是很好,更加适用于连通性比较强的物体分割以及多类别分割。
2.HRNet-seg存在一个问题,最后输出的featuremap分别是[1/4, 1/8, 1/16, 1/32],虽然是有不断的高低分辨率的交互,但是1/4还是有点捉襟见肘,会影响一些小的pixel,空洞以及边缘效果。所以做了简单的修该如下:
FPN+upsmaple形式