Tensorflow-SSD测试及训练自己的数据集

最新推荐文章于 2025-10-11 16:24:24 发布

原创

最新推荐文章于 2025-10-11 16:24:24 发布 · 2.5w 阅读

45 ·

CC 4.0 BY-SA版权

文章标签：

#SSD #python #训练自己数据

本文档介绍了如何使用Python和Tensorflow实现SSD对象检测模型，并详细阐述了从安装测试到训练自己数据集的步骤。包括软硬件配置、SSD Notebook测试、Python脚本测试，以及数据转换和模型训练过程，同时提到了可能遇到的问题及其解决方案。

一、## 软件 ##
Python + Tensorflow + OpenCV3

二、## 安装测试 ##
1、ssd_notebook.ipynb测试
（1）下载程序包并解压。源代码GitHub： balancap/SSD-Tensorflow
（2）打开终端，进入目录路径，运行：
jupyter notebook notebooks/ssd_notebook.ipynb

2、py脚本测试
不使用notebook方法，直接在python中实现。
（1）在根目录下新建一个Python脚本文件 ssd_Python.py，将ssd_notebook.ipynb中的代码复制进去，其中将最后几行代码进行更改，将
img = mpimg.imread(path + image_names[-5])
替换为：

for it  in image_names:
    img = mpimg.imread(path + it)
    i+=1
    if i>4: break
    rclasses, rscores, rbboxes =  process_image(img)
    visualiz