Sulince / SSD_ResNet

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

简介

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

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

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

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

About

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


Languages

Language:Python 92.3%Language:Jupyter Notebook 7.7%