SSD-tensorflow-1 做简易识别

用最简单的已训练好的模型对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为你想要进行测试的图片目录(代码中只对该目录下的最后一个文件夹进行测试,如果要想测试多幅图片或者做成视频的方式,需大家自行修改代码)

猜你喜欢

转载自www.cnblogs.com/pacino12134/p/10380733.html
今日推荐