Tensorflow对象检测API传输学习未启动

2024-04-26 06:36:24 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用Tensorflow对象检测API对一些自定义数据集进行传输学习。在

当启动训练命令时,训练从未开始,脚本似乎永远在运行。以下是一些输出:

WARNING:tensorflow:From /Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/tensorflow/contrib/slim/python/slim/learning.py:736: __init__ (from tensorflow.python.training.supervisor) is deprecated and will be removed in a future version.
Instructions for updating:
Please switch to tf.train.MonitoredTrainingSession
INFO:tensorflow:Restoring parameters from /Users/clementjoudet/Desktop/dev/teeth/models/transfer/ssd_mobilenet_v2_coco_2018_03_29/model.ckpt
INFO:tensorflow:Restoring parameters from /Users/clementjoudet/Desktop/dev/teeth/models/transfer/ssd_mobilenet_v2_coco_2018_03_29/model.ckpt
INFO:tensorflow:Starting Session.
INFO:tensorflow:Starting Session.
INFO:tensorflow:Saving checkpoint to path /Users/clementjoudet/Desktop/dev/teeth/models/transfer/train/model.ckpt
INFO:tensorflow:Saving checkpoint to path /Users/clementjoudet/Desktop/dev/teeth/models/transfer/train/model.ckpt
INFO:tensorflow:Starting Queues.
INFO:tensorflow:Starting Queues.
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:global_step/sec: 0

我尝试改变Tensorflow版本,从模型repo中提取,并使用不同的模型(fastrnn、SSD_mobilenet),但总是一样的。在

我正在使用https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md中的模型

以及来自https://github.com/tensorflow/models/tree/master/research/object_detection/samples/configs的同名配置文件

我在配置文件中更改的唯一内容是:

  • 类数(我的数据集的类较少)
  • 微调检查点路径
  • 列车和评估记录的输入路径+标签地图路径

注:当我在没有微调检查点的情况下开始训练时,训练效果良好。当试图从tensorflow动物园导入模型时,它是不起作用的

这是我的文件夹树:

^{pr2}$

Tags: devinfomodelmodelstensorflowstepsecglobal