ssd_mobilenet_v1的fine-tuning成果展示

ssd_mobilenet_v1的fine-tuning成果展示_第1张图片

 

 

ssd_mobilenet_v1的fine-tuning成果展示_第2张图片

 

总结:

fine-tuning的模型为ssd_mobilenet_v1_coco_2018_01_28

训练的步数为50000 steps

验证的步数为2000 steps

训练的数据集为VOCtrainval_11-May-2012

整个训练流程为:

1. 下载VOCtrainval_11-May-2012数据集

2. 将VOCtrainval_11-May-2012数据集转换成pascal_train.record和pascal_val.record两个文件

3. 下载ssd_mobilenet_v1_coco_2018_01_28模型

4. 改写pipeline.config文件

5. 开始训练

note:新版本的models已经把train.py移动到object_detection/legacy/文件夹下,可用object_detection下的model_main.py来训练

 

 

 

你可能感兴趣的:(ssd_mobilenet_v1的fine-tuning成果展示)