SSD-Tensorflow
声明:
更详细的代码解读.
欢迎关注.
运行环境:
Python3 + Tensorflow1.5 + OpenCV-python3.3.1 + Numpy1.13
windows和ubuntu环境都可以
准备工作:
请在下载模型,并放到SSD_model文件夹下
文件说明:
1、ssd300_vgg.py:搭建ssd300网络模型,并解码网络输出
·ssd_layers.py:SSD中各种网络层的定义
·ssd_anchors.py:SSD的先验框anchor设置,与Caffe源码保持一致
2、utils.py:功能函数:
(1)预处理图片:白化、resize300x300、增加batchsize这一维度
(2)处理/筛选边界框:
·将边界框超出图片范围(0,0)-(300,300)的部分cut掉;
·按类别置信度scores降序,对边界框进行排序并仅保留t
2022-04-14 10:43:04
1.13MB
Python
1