TF+SSD环境搭建

Posted 2021-02-18 thgpddl

TensoeFlow+SSD的环境搭建

1. 先下载TensorFlow的models,然后进入models_master文件夹

https://github.com/tensorflow/models

2. 安装依赖库(见3~)

Protobuf 3.0.0
Python-tk
Pillow 1.0
lxml
tf Slim (which is included in the "tensorflow/models/research/" checkout)
Jupyter notebook
Matplotlib
Tensorflow (1.15.0)
Cython
contextlib2
cocoapi

3. 对于TF有CPU和GPU版本

pip3 install tensorflow
pip3 install tensorflow-gpu

4. 其他依赖库

sudo apt-get install protobuf-compiler python-tk
pip install --user Cython
pip install --user contextlib2
pip install --user pillow
pip install --user lxml
pip install --user jupyter
pip install --user matplotlib

5. COCO API

git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
make(这是编译指令,需要我们先将Makefile文件中的Python修改为python3)
cp -r pycocotools <path_to_tensorflow>/models/research/