TensorFlow object_detect ssd_mobilenet_v1 如何预加载模型进行finetune以及不预加载模型重新训练

TensorFlow object_detect预加载模型进行finetune:

TensorFlow object_detect API预加载模型需要在训练的配置文件中配置预加载模型的路径等参数如下:

在train_config中配置

  fine_tune_checkpoint: "xxxx/xxxx/model.ckpt-90000"  #预加载的模型路径

  from_detection_checkpoint: true

TensorFlow object_detect ssd_mobilenet_v1 如何预加载模型进行finetune以及不预加载模型重新训练

预加载的模型文件必须与自己定义的模型结构保持一致,否则会出现模型未加载成功的warning提示。

 

TensorFlow object_detect如何不预加载模型,重新开始训练:

若不想加载模型,对目标检测模型重头开始训练那么只需要注释上述两个参数即可。