Technology-Object Detect 训练

发表于 | 分类于 | |

基于 Tensorfow 的 Object Detect 训练。

1
2
3
4
5
6
7
python xml_to_csv.py

python generate_tfrecord.py --csv_input=data/train_labels.csv --output_path=data/train.record

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

python export_inference_graph.py --input_type image_tensor --pipeline_config_path training/ssd_mobilenet_v1_pets.config --trained_checkpoint_prefix training/model.ckpt-数字 --output_directory halmet_inference_graph