文章自用,如有疑问请留言
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