YOLOv3-model-pruning

用 YOLOv3 模型在一个开源的人手检测数据集 oxford hand 上做人手检测，并在此基础上做模型剪枝。对于该数据集，对 YOLOv3 进行 channel pruning 之后，模型的参数量、模型大小减少 80% ，FLOPs 降低 70%，前向推断的速度可以达到原来的 200%，同时可以保持 mAP 基本不变（这个效果只是针对该数据集的，不一定能保证在其他数据集上也有同样的效果）。

环境

Python3.6, Pytorch 1.0及以上

YOLOv3 的实现参考了 eriklindernoren 的 PyTorch-YOLOv3 ，因此代码的依赖环境也可以参考其 repo

数据集准备

下载widerface数据集，得到压缩文件（提取码: ymx2）
将压缩文件解压到 Dataset

执行 widerface_label.py，生成 images、labels 文件夹和 train.txt、valid.txt 文件

剪枝算法介绍

本代码基于论文

yolo3训练人脸检测模型相关推荐

WiderFace数据集用于训练人脸检测模型
下载链接: http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/ 该数据集可用于进行人脸检测模型的训练如果做训练,可以点击网页中的 (国内还是百度吧,会快一 ...
目标检测 YOLO v3 训练人脸检测模型
YOLO,是You Only Look Once的缩写,一种基于深度卷积神经网络的物体检测算法,YOLO v3是YOLO的第3个版本,检测算法更快更准. 本文源码:https://github.com ...
百度aistudio上使用yolov4训练人脸检测模型
前几天在百度的aistudio上使用yolov4训练了一下人脸检测的模型,今天看见有人留言希望我写一篇在aistudio上如何使用yolov4的教程,纠结了一下,因为这玩意的操作流程其实在我那篇yol ...
使用卷积神经网络（CupCnn）训练人脸检测模型
结果展示最近尝试做了下人脸检测,先上两张效果图吧: 这已经是跳出来的效果比较好的图片了,表现的确是有点差. 我用了4000(19*19)多张人脸图片和8000(19*19)多张非人脸图片训练出了这个 ...
使用opencv训练目标检测模型基于cascade模型
使用opencv训练目标检测模型基于cascade模型基于Haar特征的cascade分类器(classifiers) 是Paul Viola和 Michael Jone在2001年,论文" ...
模型仅1MB，更轻量的人脸检测模型开源，效果不弱于主流算法
乾明编辑整理量子位报道 | 公众号 QbitAI AI模型越来越小,需要的算力也也来越弱,但精度依旧有保障. 最新代表,是一个刚在GitHub上开源的中文项目:一款超轻量级通用人脸检测模型. ...
大小仅1MB，超轻量级通用人脸检测模型登上GitHub趋势榜
机器之心报道项目作者:Linzaer 近日,用户 Linzaer 在 Github 上推出了一款适用于边缘计算设备.移动端设备以及 PC 的超轻量级通用人脸检测模型,该模型文件大小仅 1MB,320 ...
3模型大小_Github推荐一个国内牛人开发的超轻量级通用人脸检测模型
Ultra-Light-Fast-Generic-Face-Detector-1MB 1MB轻量级通用人脸检测模型作者表示该模型设计是为了边缘计算设备以及低功耗设备(如arm)设计的实时超轻量级通用 ...
DBFace升级，模型仅1.3M的轻量级高精度人脸检测模型
本文转载自DeepBlue深兰科技已获原文作者授权转载,请勿二次转载前段时间DBFace人脸检测库横空出世,被高度评价为:"这个仅7M大小的人脸识别模型几乎识别出了世界最大自拍照中的所有 ...

yolo3训练人脸检测模型

YOLOv3-model-pruning

环境

数据集准备

剪枝算法介绍

yolo3训练人脸检测模型相关推荐

最新文章

热门文章