MTCNN RetinaFace YoloFace

chencao100

已于 2022-12-06 10:43:09 修改

阅读量1.3k

点赞数

分类专栏： nn 文章标签： mtcnn

于 2019-08-15 15:44:39 首次发布

本文链接：https://blog.csdn.net/chencao100/article/details/99639315

版权

nn 专栏收录该内容

10 篇文章

订阅专栏

MTCNN
Fast-MTCNN /mtcnn_opencv.cpp
bug:
line 174 faceBox.score = confidence_data[i];
should be : faceBox.score = 1.0f - confidence_data[i];

RetinaFace-Cpp

retinaface
RetinaFace-Cpp
retinaface_caffe
Retinaface-caffe

scrfd

修改mmdet\__init__.py

mmcv_minimum_version = '1.1.5'
mmcv_maximum_version = '1.4.8'

安装mmdet

python setup.py develop

导出onnx

python tools/scrfd2onnx.py configs/scrfd/scrfd_34g.py model.pth --shape 640 640 --input-img t2.jpg

训练修改 configs/scrfd/scrfd_34g.py

	total_epochs = 10*lr_mult 	#80
	checkpoint_config = dict(interval=10)	#80
	evaluation = dict(interval=10, metric='mAP')	#80
    samples_per_gpu=1, 	#8
    workers_per_gpu=1, 	#3

修改 mmdet\models\dense_heads\scrfd_head.py

        avg_factor = sum(avg_factor)
        avg_factor = reduce_mean(avg_factor).item()
        if  avg_factor != 0:	#+++
            losses_bbox = list(map(lambda x: x / avg_factor, losses_bbox))
        losses = dict(loss_cls=losses_cls, loss_bbox=losses_bbox)
        if self.use_kps:
            if  avg_factor != 0:	#+++
                losses_kps = list(map(lambda x: x / avg_factor, losses_kps))
            losses['loss_kps'] = losses_kps
        if self.use_dfl:
            if  avg_factor != 0:	#+++
                losses_dfl = list(map(lambda x: x / avg_factor, losses_dfl))
            losses['loss_dfl'] = losses_dfl
        return losses

python tools/train.py configs/scrfd/scrfd_34g.py  --gpus 1

scrfd-opencv

YoloFace

yolov5
yolov5-face
yolov7
yolov7-face
opencv Support for YOLOv7 ONNX
修改 dnn\src\layers\fast_convolution\fast_convolution.cpp

        int k = 0, nbias = K + 32;
        conv->biasBuf.resize(nbias);//reserve //20221125 opencv-4.x