github链接:https://github.com/georgesung/ssd_tensorflow_traffic_sign_detection、
基于SSD网络做的交通标志识别,实现了stop和pedestrian crossing的识别,前置网络由SSD原文的VGG改为了Alexnet
模型权重文件下载
首先需要下载作者训练好的权重文件(需要合理上网才可以下载),直接放到根目录即可(直接将后缀为 .ckpt的三个checkpoints文件放到download下来的项目文件ssd_tensorflow_traffic_sign_detection-master文件下即可)
【tensorflow_error】‘module’ has no attribute ‘select’ 问题
这个是由于TensorFlow 版本API发生变化引起,直接将model.py文件中的tf.select
改为tf.where即可解决。
python TypeError: Expected int32, got list containing Tensors of type ‘_Message’ instead.
tensorflow1.0及以后api定义:(数字在后,tensors在前)
需要将 tf.concat() 中的 数字与tensor元素对调即可解决