SSD: Single Shot MultiBox Detector
论文链接:https://arxiv.org/pdf/1512.02325.pdf
Pytorch代码:https://github.com/shanglianlm0525/PyTorch-Networks
Pytorch代码:
import torch
import torch.nn as nn
import torchvision
import cv2
def Conv3x3BNReLU(in_channels,out_channels,stride,padding=1):
return nn.Sequential(
nn.Conv2d(in_channels=in_channels, out_channels=out_channels, kernel_size=3, stride=stride, padding=padding),
nn.BatchNorm2d(out_channels),
n