Tensorflow 训练自己的目标检测与识别模型(四)
前面对数据集的创建以及图像数据增强进行了叙述,见链接:https://blog.csdn.net/weixin_41644725/article/details/85678348 和 https://blog.csdn.net/weixin_41644725/article/details/85687049
接下来对项目中的配置进行叙述:
(1)下载所需要的预训练模型。
见链接:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
如图下图所示。
下载好模型之后,解压该文件之后得到3个model.ckpt文件,将其考到存放预训练模型的文件夹下,如下图所示。