faster rcnn ubuntu环境配置说明

文章自用,如有疑问请留言

wget https://github.com/tensorflow/models/archive/master.zip

unzip master.zip

apt-get -y install protobuf-compiler

cd models-master/research/

protoc object_detection/protos/*.proto --python_out=.

python3 setup.py install

cd ./slim;python3 setup.py install

cd ..

python3 object_detection/builders/model_builder_test.py

python3 object_detection/train.py --logtostderr --pipeline_config_path=/content/demo/ssd_mobilenet_v1_pets.config --train_dir=/content/demo/savemodel

python3 ./models-master/research/object_detection/export_inference_graph.py  --input_type image_tensor  --pipeline_config_path /content/demo/ssd_mobilenet_v1_pets.config  --trained_checkpoint_prefix /content/demo/savemodel/model.ckpt-806 --output_directory /content/demo

posted @ 2018-03-28 09:59  乐哈哈1  阅读(452)  评论(0编辑  收藏  举报