![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
RetinaFace
王小凡wxf
这个作者很懒,什么都没留下…
展开
-
pytorch-----RetinaFace(代码网络理解)
1、网络模型2、训练代码在之前的博客中。(1)骨干网:mobilenet0.25输入图片[640,640,3]batchsize=32有三层输出:对应图片的c3,c4,c5;out1[32,64,80,80]out2[32,128,40,40]out3[32,256,20,20]特征金字塔:fpn1[32,64,80,80]fpn2[32,64,40,40]fpn3[...原创 2020-04-18 19:44:13 · 933 阅读 · 0 评论 -
pytorch----retinaface(三种损失使用的函数)
pytorch----retinaface(三种损失使用的函数)retinaface损失函数定义:(1)L(cls)(面分类损失):#计算交叉熵loss_c = F.cross_entropy(conf_p, targets_weighted, reduction='sum')1、交叉熵的公式:其中P为真实值,Q为预测值。2、计算交叉熵的步骤:1)步骤说明:①将predic...原创 2020-04-12 18:46:42 · 1855 阅读 · 0 评论 -
pytorch----retinaface(测试)
test_widerface.pyfrom __future__ import print_functionimport osimport argparseimport torchimport torch.backends.cudnn as cudnnimport numpy as npfrom data import cfg_mnet, cfg_re50from layers.f...原创 2020-04-10 22:36:01 · 630 阅读 · 0 评论 -
pytorch----retinaface(训练)
pytorch----retinaface(训练)train.pyfrom __future__ import print_functionimport osimport torchimport torch.optim as optimimport torch.backends.cudnn as cudnnimport argparseimport torch.utils.data...原创 2020-04-08 11:05:03 · 2058 阅读 · 3 评论 -
pytorch----retinaface(data文件)
data文件目录FDDB文件存放fddb数据集:这里不做说明。config.py:# config.pycfg_mnet = { 'name': 'mobilenet0.25', 'min_sizes': [[16, 32], [64, 128], [256, 512]], 'steps': [8, 16, 32],##步幅与论文中有所不同 'var...原创 2020-04-07 21:48:42 · 636 阅读 · 2 评论 -
pytorch----retinaface(layer文件)
pytorch----retinaface(layer文件)(改了SSD的源码,添加了mandl)目录prior_box.pyimport torchfrom itertools import product as productimport numpy as npfrom math import ceil#生成先验框class PriorBox(object): d...原创 2020-04-07 17:31:30 · 307 阅读 · 0 评论 -
pytorch----RetinaFace(models)
自己的一些理解,如有错误请各位大老指出。models文件文件目录models为python文件自带__init__.py。net.py:import timeimport torchimport torch.nn as nnimport torchvision.models._utils as _utilsimport torchvision.models as models...原创 2020-04-04 18:25:59 · 544 阅读 · 0 评论 -
pytorch----RetinaFace(utils文件)
自己的一些理解,如有错误请各位大老指出。utils文件文件目录nmsnms为python文件自带__init__.pypy_cpu_nms.py代码为:# --------------------------------------------------------# Fast R-CNN# Copyright (c) 2015 Microsoft# Licensed und...原创 2020-03-31 21:46:27 · 377 阅读 · 1 评论