我的CV实习工作总结

本文是作者在CV实习期间的工作总结,主要涉及人脸检测数据的清洗和模型训练。通过调用百度API与OpenPose进行数据预处理,采用Yolov3模型并进行剪枝优化,实现了高效人脸检测。最后,开发了SDK以支持批量检测和结果可视化。
摘要由CSDN通过智能技术生成

两个月不到的实习,总结一下。

洗数据的大路走到了黑

我负责的模块是人脸检测,人脸检测的最终目的,是可以实现视频中人脸实时的关键点检测。对于检测的recall,precision,检测框是否切脸,检测速度都有要求。
需要解决的问题是:无法检测出靠近镜头的大脸图片,易对背景产生误检
解决办法:最好的解决办法就是给高质量的数据集(很无奈哦)
然后开始洗数据
方法:

  1. 首先调用了百度API,这里写了一个调用脚本,然后做了本地可视化,评估了一下几个指标。百度的人脸检测严丝合缝,人脸是歪的,那么检测框就是歪的,我特地让检测框的坐标做了仿射变换,然后按照检测框(w,h)和旋转角度做了一个权重运算,使得检测框可以方方正正,然后又不切脸;
  2. 然后调用本地openpose,也可视化了一下,评估指标。
    骚操作来了!
  3. 把两个label做了融合:取两个label的交集!这样会降低召回,但是大大增加了准确率,降低了切脸率。具体做法是计算IOU,选IOU最大值作为选取的检测框。这里附一下我的代码,但是由于没有openpose的模块,所以大家可以看看流程
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值