用最简单的已训练好的模型对20类目标做检测。
你电脑的tensorflow + CUDA + CUDNN环境都是OK的, 同时python需要安装cv2库
{ 'aeroplane' 'bicycle' 'bird' 'boat' 'bottle' 'bus' 'car' 'cat' 'chair' 'cow' 'diningtable' 'dog' 'horse' 'motorbike' 'person' 'pottedplant' 'sheep' 'sofa' 'train' 'tvmonitor' }
下载好代码后,在checkpoints文件夹下面由ssd_300_vgg.ckpt,直接解压到当前文件夹。
找到notebooks文件夹里的ssd_notebook.ipynb,用jupyter打开。
将读入的图片改为自己的图片就行了(更改path或者将自己的图片放到demo文件夹下面),然后运行所有cells。
- path为你想要进行测试的图片目录(代码中只对该目录下的最后一个文件夹进行测试,如果要想测试多幅图片或者做成视频的方式,需大家自行修改代码)