insightface项目Retinaface训练方法
【背景】最近由于公司项目需要,花了点时间入门了一下深度学习算法。选择deepinsight/insightface项目的Retinaface算法练手,按照项目说明的training步骤无法直接运行,这里把踩过的坑都记录一下。
#下载源码:
https://github.com/deepinsight/insightface.git
#环境部署
这里用的是MXNet的框架,MXNet和CUDA的环境部署不在本文档的范围内,大家自行百度,我的环境是ubuntu 18.04.3\python 3.7\CUDA 10\cudnn7.5,硬件是一块GTX 1080,显卡驱动是nvidia-driver-430(由于显卡算力有限,我没有训练处自己的模型,但是不影响本文的介绍的整体流程)
第二步我跳过不执行
第三步期间会要求安装一些依赖,用pip安装,直接用原生的源地址就可以了。
#数据准备
第一步很重要,要用作者自己准备的标注,不能用widerface自己的标注。下载以后解压出来里面有三个文件夹test、train、val,把train路径里面的label.txt复制到第三步的路径里面。
第二步就是下载,原来的数据集是再百度云,现在迁移到了腾讯云,没差,对应没有会员的人来说都一样慢。这里要用官方的版本,有些版本把所有的图片都放到了一个目录下面,也无法训练。
第三步要注意是在RetinaFace文件夹路径下再创建data文件夹,下图是我的文件路径。
#训练
前两步没什么需要强调的问题,我再第步运行的时候遇到两个问题,第一个是再Pycharm里面,设置环境变量没有生效,所以我直接再代码里面修改了使用的GPU,再train.py的359行,直接改成了cvd = '0,1,2'.strip(),其中0, 1, 2指的是你训练时候要用的GPU。
第二个问题是启动配置里面没有OHEM_MODE的默认配置,所以导致执行到下面内容的时候抛出异常:
这里我直接在rcnn/config.py加了config.TRAIN.OHEM_MODE = 1。但是我并不明白OHEM算法,也不知道这个模式选择对训练结果有什么影响,希望知道的小伙伴在评论中帮忙解答一下。
接下来就是漫长的训练过程了,由于设备性能有线,要完成10000轮 WIDERFACE 数据集的训练,需要233天....所以只能先学习到这里了,等借到了更强劲的设备以后再把模型训练出来试试效果如何~
insightface项目Retinaface训练方法相关推荐
- Insightface项目爬坑指南+使用本地数据集训练流程(MXNET版)
其实半年多前就已经把insightface训练等一系列环节弄熟了,不得不说IBUG组的这个模型确实是开源界的翘楚,但是还是存在一些问题在某些程度上和商汤云从等大厂存在一点差距,这不妨碍大部分人日常人脸 ...
- 人脸生成对抗+人脸识别流程+insightface
一.人脸生成对抗 传统的DeepFake 需要大量的原始数据,并且要经过好几天的训练才能达到高质量的效果. 黎颢团队提出,将DeepFake和他此前做的怕GAN结合到一起,做了一个新系统.不需要大量训 ...
- RV1126_python人脸识别Retinaface+MobilefaceNet
RV1126_python人脸识别Retinaface+MobilefaceNet RV1126 具备RKNN 模块支持大部分如Pytorch.MXNet.Caffe.tensorflow.keras ...
- 开源大佬面对面:InsightFace带你打造开箱即用的人脸识别系统
导读 提到人工智能和人脸识别开源技术,相信很多业内的同行对于InsightFace应该不陌生,其中的Arcface论文被引用超过2100+,自从2018年开源以来,在GitHub上的增长也保持着强劲的 ...
- retinaface查看样本
开源项目retinaface的标注查看脚本 https://github.com/biubug6/Pytorch_Retinaface import osimport cv2if __name__ = ...
- @property python知乎_使用Mxnet进行图像深度学习训练工具 InsightFace - 使用篇, 如何一键刷分LFW 99.80%, MegaFace 98%....
开头先把论文和开源项目地址放一下: Additive Angular Margin Loss for Deep Face Recognitionarxiv.org deepinsight/insig ...
- InsightFace实现人脸比对[升级版] | 不用建立索引的那种 (二)
InsightFace-paddle的版本还是太麻烦了 并没有真正做到不建立索引 下面来说一下真正不需要建立索引的 https://github.com/deepinsight/insightface ...
- GitHub Star破万!InsightFace:一个2D和3D深度人脸分析工具箱
导读 提到人工智能和人脸识别开源技术,相信很多业内的同行对于InsightFace应该不陌生,其中的Arcface论文被引用超过2100+,自从2018年开源以来,在GitHub上的增长也保持着强劲的 ...
- 最强六大开源轻量级人脸检测项目分析
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 随着深度学习的兴起,工业界和学术界越来越多的使用基于深度学习的方法 ...
最新文章
- 学习C语言必须知道的理论知识(第一章)
- 三方会谈进行时,Uber旧金山无人车已被叫停
- HTML地址栏传数据和json区别,前端利用formData格式进行数据上传,前端formData 传值 和 json传值的区别?...
- Spring基础知识和配置
- 台式电脑耳机插孔在哪_不到一千元的迷你电脑究竟如何?Intel NUC7CJYH测评来了...
- ITK:图像的拉普拉斯算子
- Linux平台常用命令
- 良心发现,时隔一年再回首莫比乌斯反演(公式性质证明+题目练习)
- 【FFMPEG系列】之打印调试
- 开源视频监控系统:iSpy
- 【渝粤教育】国家开放大学2018年秋季 1301T病理生理学 参考试题
- 程序员35岁之后的职业发展
- PHP+Swoole 搭建 Websocket 聊天室
- Latex表格/公式在线编译转换器
- ciscn 2020 Misc the_best_ctf_game
- banner图片自适应
- 悟空云课堂|第四十七期:会话固定(CWE-384: Session Fixation)
- 数据包嗅探 Packet Sniffing
- win10在电脑中添加新用户
- 有道云笔记linux使用教程,办公必备神器系列五:云笔记:为知笔记、印象笔记、有道云笔记...