Pytorch 搭建yolov3 搭建yolov3的网络结构
yolov3的网络结构如下图
这张图片转自此篇博客.
观察我们可以发现Darknet53本身由大量的残差连接构成,所以首先我们实现基础卷积模块
创建基础卷积模块DBL
########################
# 创建一个基础的卷积模块 DBL
########################
class DBl(Module):
def __init__(self, inplane, plane, kernel_size=1, padding=0, stride=1):
原创
2021-02-10 11:31:32 ·
662 阅读 ·
0 评论