一、环境配置
基本环境:Windows 10 + GTX950M
1、安装Anaconda3()
注意:必须下载Anaconda3,因为Anaconda3对应Python3.x,而Windows下Tensorflow只支持Python3.x版本。
2、安装Python-OpenCV3
然后,打开cmd,进入下载文件所在目录,运行:
pip install opencv_python-3.2.0-cp35-cp35m-win_amd64.whl
二、SSD-Tensorflow演示
git clone https:
或者直接点击“Download ZIP”手动下载
2、打开cmd进入SSD-Tensorflow根目录下的./checkpoint,解压“ssd_300_vgg.ckpt.zip”
然后,再进入SSD-Tensorflow的根目录下运行
jupyter notebook notebooks/ssd_notebook.ipynb
即可。
但是,可能因为我的系统还装有Anaconda2的原因,直线失败,所以我将ssd_notebook.ipynb中的代码复制出来,在notebooks文件夹中新建了一个Python脚本文件,并且将ssd_notebook.ipynb中的最后几行代码改了下
将
img = mpimg.imread(path + image_names[-5])
后面的几句改为:
for it in image_names:
img = mpimg.imread(path + it)
rclasses, rscores, rbboxes = process_image(img)
visualization.plt_bboxes(img, rclasses, rscores, rbboxes)
然后在cmd中进入notebooks直接运行
Python3 ssd_Python.py
即可运行。
由于网速问题,图片暂时上传不了。
三、SSD-Tensorflow训练
1、基于某个训练节点上对网络进行微调
2、基于已训练好的网络进行微调
未完待续….