本文所使用的数据和模型可在优快云资源页下载。
链接:
网络定义文件
数据链接和检测用lst文件
本文主要对原来的代码进行了整理,方便调用和训练。
主要参考了Gluon的SSD例子。
- SSD网络模型定义
ssd.py
import mxnet as mx
import matplotlib.pyplot as plt
import os.path as osp
import mxnet
本文所使用的数据和模型可在优快云资源页下载。
链接:
网络定义文件
数据链接和检测用lst文件
本文主要对原来的代码进行了整理,方便调用和训练。
主要参考了Gluon的SSD例子。
ssd.py
import mxnet as mx
import matplotlib.pyplot as plt
import os.path as osp
import mxnet