参考链接https://github.com/tensorflow/models/blob/master/object_detection/g3doc/running_pets.md
先参考https://github.com/tensorflow/models/blob/master/object_detection/g3doc/installation.md安装好环境
以下默认都在models目录下操作
mkdir petstrain
注意PYTHONPATH库路径的设置
# From tensorflow/models/ export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
下载Oxford-IIIT Pets的图片和标注信息
# From tensorflow/models/cd petstrain wget http://www.robots.ox.ac.uk/~vgg/data/pets/data/images.tar.gz wget http://www.robots.ox.ac.uk/~vgg/data/pets/data/annotations.tar.gz tar -xvf images.tar.gz tar -xvf annotations.tar.gzcd ..
转换成TFRecord格式
# From tensorflow/models/cp object_detection/data/pet_label_map.pbtxt petstrain/python3 object_detection/create_pet_tf_record.py --label_map_path=petstrain/pet_label_map.pbtxt --data_dir=petstrain --output_dir=petstrain
在petstrain目录生成pet_train.record pet_val.record
下载预训练的模型
cd petstrainwget http://storage.googleapis.com/download.tensorflow.org/models/object_detection/faster_rcnn_resnet101_coco_11_06_2017.tar.gz tar -xvf faster_rcnn_resnet101_coco_11_06_2017.tar.gzmv faster_rcnn_resnet101_coco_11_06_2017/model.ckpt.* .cd ..
修改配置文件
cp object_detection/samples/configs/faster_rcnn_resnet101_pets.config petstrain/
# From tensorflow/models/ # Edit the faster_rcnn_resnet101_pets.config template. Please note that there # are multiple places where PATH_TO_BE_CONFIGURED needs to be set. sed -i "s|PATH_TO_BE_CONFIGURED|petstrain|g" petstrain/faster_rcnn_resnet101_pets.config
训练的目录下会有以下文件
+ petstrain/
- faster_rcnn_resnet101_pets.config
- model.ckpt.index
- model.ckpt.meta
- model.ckpt.data-00000-of-00001
- pet_label_map.pbtxt
- pet_train.record
- pet_val.record开始训练
# From tensorflow/models/python3 object_detection/train.py --logtostderr --pipeline_config_path=petstrain/faster_rcnn_resnet101_pets.config --train_dir=petstrain/res/
查看训练进度
# From tensorflow/models/tensorboard --logdir=pet-train/res/
打开浏览器localhost:6006
导出训练好的图
# From tensorflow/models/ object_detection/export_inference_graph.py --input_type image_tensor \ --pipeline_config_path petstrain/faster_rcnn_resnet101_pets.config \ --checkpoint_path petstrain/res/model.ckpt-445 \ --inference_graph_path petstrain/output_inference_graph.pb
这样便得到output_inference_graph.pb文件
时间： 2024-10-12 04:35:01