Tensorflow：训练预训练模型

Question

有人知道如何从Tensorflow动物园训练预训练模型吗？ 我尝试使用不同的模型和不同的tf版本（1.8-1.11）。

最终，我遇到了与此错误类似的错误：

  

NotFoundError（请参阅上面的回溯）：关键转换/偏见/动量没有   在检查点[[Node：save / RestoreV2 =   RestoreV2 [dtypes = [DT_FLOAT，DT_FLOAT，DT_FLOAT，DT_FLOAT，DT_FLOAT，   ...，DT_FLOAT，DT_FLOAT，DT_FLOAT，DT_FLOAT，DT_INT64]，   _device =“ / job：localhost /副本：0 / task：0 / device：CPU：0”]（_ arg_save / Const_0_0，   保存/恢复V2 / tensor_names，保存/恢复V2 / shape_and_slices）]]

我刚从刚加载的模型中使用了一个检查点。在pipelineconfig中，我仅更新“ PATH_TO_BE_CONFIGURED”。

是模型动物园还是我的环境/数据集/ tfrecords的问题？

有人有一个简单的例子吗？

更新

我使用“ LabelImg”来注释bbox。

然后我使用this script来生成包括TFRecords在内的Train＆Eval数据集。

我尝试了TF模型动物园中的以下模型：

• faster_rcnn_resnet101_coco_2018_01_28
• faster_rcnn_resnet101_kitti_2018_01_28
• faster_rcnn_inception_resnet_v2_atrous_oid_2018_01_28

此处是“ {faster_rcnn_inception_resnet_v2_atrous_oid_2018_01_28”的pipeline config。

这里是the script，我用它来训练模型。

如果我删除除config之外的所有模型文件，那么任何模型都将得到很好的培训。但是，如果尝试使用检查点-我得到了错误：|

我当前的TF版本：1.8.0

操作系统：Ubuntu 18.04.2 LTS