Ubuntu16.04下下tensorflow+SSD实现(人脸,通用物体)目标检测,环境的搭建与实现(人脸,通用物体)目标检测,环境的搭建与
测试以及搭建环境的踩坑环节的梳理(一)测试以及搭建环境的踩坑环节的梳理(一)
Ubuntu16.04下下tensorflow+SSD实现(人脸,通用物体)目标检测,环境的搭建与测试以及搭建环境的踩坑环节的梳理(一)实现(人脸,通用物体)目标检测,环境的搭建与测试以及搭建环境的踩坑环节的梳理(一)
开发环境:Ubuntu16.04、cuda、cudnn
语言:Python3.5(也可以Python3.6)
框架:tensorflow1.12 、SSD
英伟达GPU配置步骤:
1.tensorflow1.12使用的Python版本不能是3.7的,要求在3.7以下
2.cuda和cudnn的环境在英伟达下进行下载,由于每个人的计算机环境是否有GPU显卡和显卡的版本不同,需要进行TensorFlow和cuda、cudnn和匹配
2.1 nvidia-smi 可以先用这个命令看看GPU是否存在
2.2 sudo apt-get update 此次可以查看到你可以安装的nvidia驱动版本
2.3 安装驱动 可以在系统设置的附加驱动中选择合适自己的显卡驱动,注意
是要英伟达的,版本经量选择低的
2.4 安装cuda:https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604
2.5 安装cudnn:https://developer.nvidia.com/cudnn 需要注册个社区账号,我好像是微信登录的~
安装好后进行cuda和cudnn的环境的配置(英伟达有有教程)再次用nvida-smi就能看到GPU的信息了
SSD模型的配置:
1.在TensorFlow社区下载models,但是TensorFlow里的模型进行过针对TensorFlow2.0的模型调整,但是在测试文件中有些代码是TensorFlow1.XX版本的,(例如:其
中tf.contrib模块在tensorflow2.0中被弃用了)所以我下载的是旧版的SSD :models
新版的是:models-master。虽然新的也能进行模型训练,但是我做不了环境的搭建测试会报错,模型下载链接
:https://github.com/tensorflow/models
2.下载好模型后可以参照github的安装步骤进行安装,这里的安装步骤没有进行更新,始终是保持旧版的,但是影响不大SSD安装:
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md
这是我旧版模型旧版模型的地址和较旧版模型较旧版模型的地址,他们测试环境代码都一样,我使用的是models:
链接: https://pan.baidu.com/s/1a445D8ZdbjvJTkJvv-jg0g 提取码: sh5j
TensorFlow1.12安装-i 后面我使用的是国内的镜像源,不然下载贼慢
这里还有,我无论是pip还是Python都是使用pip3和python3,这样安装就能在Python3 的环境下安装,不然会默认安装在Python2的环境中
sudo pip3 install -i https://pypi.douban.com/simple tensorflow==1.12.0
其他依赖库的安装
pip3 install --user Cython
pip3 install --user contextlib2
pip3 install --user pillow
pip3 install --user lxml
pip3 install --user jupyter
pip3 install --user matplotlib
COCOAPI的安装,在github下载比较慢可以百度网盘,下载好后进到PythonAPI进行make,将pycocotools复制到/models/research/目录下就行了
这是我自己的:链接: https://pan.baidu.com/s/1ouvdKdMJEGC-zLjwBk94rA 提取码: 4deg
git clone https://github.com/cocodataset/cocoapi.git
评论0