1、下載models:鏈接:https://pan.baidu.com/s/1TNeg20BvkgxruqyTH_ClxA 密碼:rky0
2、下載cocoapi:https://github.com/cocodataset/cocoapi.git
git clone https://github.com/cocodataset/cocoapi.git
3、參考官方提供的文檔配置環境
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md
我是的環境是Anaconda下的,專門創建了學習tensorflow的環境,基於python3.
搭建環境要注意是在哪個版本下搭建,之前就是弄錯了結果一直不能導包,到最後一步檢查時提示沒有contrib,這是因爲默認的base環境下我沒有install tensorflow。
4、在anaconda下配置子環境時,所有的文件都在envs文件夾下,將models文件夾放在envs下的tensorflow文件夾中,因爲這裏我將子環境命名爲tensorflow
(base) appledeMacBook-Pro:~ apple$ activate tensorflow
(base) appledeMacBook-Pro:~ apple$ conda info -e
# conda environments:
#
base * /anaconda3
tensorflow /anaconda3/envs/tensorflow
可以看出默認是base環境,所以將models文件夾放在…/anaconda3/envs/tensorflow/下,同時將下載好的cocoapi放在models裏。
5、接下來是cocoapi的安裝
官方步驟是:
進入cocoapi中的PythonAPI文件夾,對裏面Makefile執行make操作:
因爲我是在tensorflow子環境中,所以在終端要切換環境:
(base) appledeMacBook-Pro:~ apple$ conda activate tensorflow
make操作:
(tensorflow) appledeMacBook-Pro:~ apple$ cd /anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow/models/cocoapi/PythonAPI
(tensorflow) appledeMacBook-Pro:PythonAPI apple$ make
makefile文件:
all:
# install pycocotools locally
python3 setup.py build_ext --inplace
rm -rf build
install:
# install pycocotools to the Python site-packages
python3 setup.py build_ext install
rm -rf build
要注意的是文件裏面使用的是python,要改成python3
make編譯之後生成了pycocotools文件夾,將這個文件夾複製到models/research/中
6、接下來步驟protoc,research文件夾下的objec_detection下的protos裏面的文件都執行了且生成相應的.py文件
添加依賴:因爲要使用slim
最後運行目標檢測裏面的文件看是否成功:
python object_detection/builders/model_builder_test.py
直接複製粘貼到終端:
即:
(tensorflow) appledeMacBook-Pro:research apple$ protoc object_detection/protos/*.proto --python_out=.
(tensorflow) appledeMacBook-Pro:research apple$ export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
(tensorflow) appledeMacBook-Pro:research apple$ python3 object_detection/builders/model_builder_test.py
..............
----------------------------------------------------------------------
Ran 14 tests in 0.080s
可以看出執行成功即配置正確。