实践目标检测--构造SSD网络与训练

网络结构图 基础网络块可自定义,可使用VGG、ResNet、DenseNet,这里使用了简单三层卷积+SSD的网络配置。 多尺度特征块:使用两层3*3的网络保持图片大小不变后,使用2*2的平均池化,将图片大小减半,来提取不同尺度的特征。 类别、边界框预测:使用多通道的输出来预测结果。 文件名SSD_model.py import sys sys.path.insert(0, '..') impo
相关文章
相关标签/搜索