/SSD_ResNet

基于SSD和ResNet的危险物品识别系统

Primary LanguagePython

简介

  这是一个基于SSDResNet的神经网络模型,将原本SSD中的VGG部分替换成ResNet。该模型可以对危险物品(例如道具、弓箭)进行检测。

  原本SSD网络模型结构如下图所示:

使用ResNet改进后的架构如下图所示:

  数据集格式采用的是VOC格式,数据集在这里下载,名称是VOC2009.zip