运行ssd_tensorflow_traffic_sign_detection遇到的各种问题以及解决办法

最新推荐文章于 2021-06-01 02:50:07 发布

weixin_42069344

最新推荐文章于 2021-06-01 02:50:07 发布

阅读量949

点赞数 1

分类专栏： Ubuntu 目标检测 SSD TensorFlow

本文链接：https://blog.csdn.net/weixin_42069344/article/details/82998384

版权

本文详述了在运行TensorFlow的SSD（Single Shot Multibox Detector）模型进行交通标志检测时遇到的问题，包括依赖安装、模块错误、API变更等，并提供了相应的解决方法。测试环境为CUDA8.0、cuDNN5.1和TensorFlow1.2.0。通过调试，最终成功运行并展示了结果。

摘要由CSDN通过智能技术生成

1.关于开源代码，读者可以根据文章标题在github上自行下载，交通标志检测

2.文章的重点。

（1）关于运行：根据代码给出的文档，首先我们要安装相关的依赖库，（需要说明的是，本文测试的环境是CUDA8.0+cuDNN5.1，操作平台Ubuntu16.04）TensorFlow版本是GPU版TensorFlow1.2.0版本，安装方法（在激活和创建环境完成的情况下，终端输入tensorflow-gpu==1.2.0，在Windows的cmd命令窗口下，是类似的操作。）因为后续出现的问题和TensorFlow版本的关系比较大，具体原因不是很清楚。故在此说明本人测试的相关环境。～～想做一个逗逼的小编，可是正经起来不是人（一副大叔心态，哎）。言归正传，哈哈，开始：相关依赖

## Dependencies
* Python 3.5+
* TensorFlow v0.12.0
* Pickle
* OpenCV-Python
* Matplotlib (optional)

这些要提前安装好，看代码开头几行导包的时候要用，这些在程序运行时候要用到的依赖和相关包

import tensorflow as tf
from settings import *
from model import SSDModel
from model import ModelHelper
from model import nms
import numpy as np
from sklearn.model_selection import train_test_split
import cv2
import math
import os
import time
import pickle
from PIL import Image
import matplotlib.pyplot as plt
from moviepy.editor import VideoFileClip
from optparse import OptionParser
import glob