前面刚写完tensorflow1.15+ssd_mobilenet_v2的环境搭建,发现tensorflow2.0+早已经支持Tensorflow object detection API,因此肯定要与时俱进,下面就开始tensorflow2.0的环境搭建吧。
首先申明,以下搭建步骤采纳同门师弟博客,文末附上原文链接,但因人而异,还是些许步骤不同,在此做个人的环境搭建总结。
conda create -n tf2 pip python=3.8
conda activate tf2 # 激活
pip install --ignore-installed --upgrade tensorflow-gpu==2.2.0 -i https://pypi.douban.com/simple/
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
git clone https://github.com/tensorflow/models.git
或者:到上述网址下载并解压,笔者就是采用这种方式,因为git clone一直下载不下来。
在激活的环境中运行命令行
conda install -c anaconda protobuf
或者下最新的压缩包,笔者这里下的是:protoc-3.13.0-linux-x86_64,进行解压
进入models/research/目录下
/home/lianlirong/models-master/protoc-3.13.0-linux-x86_64/bin/protoc object_detection/protos/*.proto --python_out=.
运行结束后可查看protos文件内,是否新生成很多.python文件
export PYTHONPATH=$PYTHON:./models-master/research/slim
方法一:
pip install cpython pip install
git+https:// github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
方法二:笔者的方法
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
make
make install
python setup.py install
cp -r ./cocoapi/PythonAPI/pycocotools ./models-master/research/ # 复制pycocotools到models-master/research/
cp object_detection/packages/tf2/setup.py .
python -m pip --default-timeout=200 install . # --default-timeout=200延长网络的反应的时间,避免下载中断
若出现:
某一依赖项一直下载不下来,则可以先中断,单独下载这一部分:
pip install tensorflow==2.3.1 -i https://pypi.douban.com/simple # 利用豆瓣源下载速度更快。
.
cd ./models-master/research
python object_detection/builders/model_builder_tf2_test.py
若出现如下结果则配置成功
就此环境搭建就好了!!!!
参考链接:
https://blog.csdn.net/l13022736018/article/details/108737082