import argparse
import time
import torch
import torch.nn as nn
import torch.nn.functional as F
# from torchvision.models._utils import IntermediateLayerGetter
from Pelee_network_v2 import PeleeNet
from Snet import Snet
# from bi_fpn import BIFPN
from bifpn import BiFPN
from darknet import DarkNet
from data import cfg_mnet, cfg_shuffle
from models.net import MobileNetV1 as MobileNetV1
from models.net import FPN as FPN
from models.net import SSH as SSH
# from models.peleenet import PeleeNet
from models.shufflenet_yolo import ShuffleNetV2
class ClassHead(nn.Module):
def __init__(self,inchannels=512,num_anchors=3,num_classes=2):
super(ClassHead,self).__init__()
self.num_anchors = num_anchors
retinaface box添加layyer
于 2020-01-20 10:51:06 首次发布
本文详细介绍了如何在RetinaFace深度学习模型中添加Box预测层,探讨了这一过程对于计算机视觉任务的重要性,特别是面部检测。通过PyTorch实现,深入解析了新增层的代码逻辑和模型结构的调整,旨在提升模型的定位精度。
摘要由CSDN通过智能技术生成