我正在尝试使用在tensorflow对象检测库中找到的eval.py脚本来评估我训练过的一些模型检查点。但是,脚本似乎挂起,无法通过以下操作:
python3 /data/tensorflow/models/research/object_detection/eval.py --logtostderr --pipeline_config_path=/home/b3020111/dolphin-recognition/pretrained/faster_rcnn_resnet50_fgvc_2018_07_19/5000_pipeline.config --checkpoint_dir=/home/b3020111/dolphin-recognition/models/faster_rcnn_resnet50_fgvc_5000_defaults/train --eval_dir=/home/b3020111/dolphin-recognition/models/faster_rcnn_resnet50_fgvc_5000_defaults/eval --gpudev=1
WARNING:tensorflow:From /home/b3020111/.virtualenvs/dolphin-detection/lib/python3.6/site-packages/tensorflow/python/platform/app.py:125: main (from __main__) is deprecated and will be removed in a future version.
Instructions for updating:
Use object_detection/model_main.py.
INFO:tensorflow:Scale of 0 disables regularizer.
INFO:tensorflow:Scale of 0 disables regularizer.
INFO:tensorflow:Scale of 0 disables regularizer.
INFO:tensorflow:depth of additional conv before box predictor: 0
INFO:tensorflow:Scale of 0 disables regularizer.
WARNING:tensorflow:From /data/tensorflow/models/research/object_detection/predictors/heads/box_head.py:92: calling reduce_mean (from tensorflow.python.ops.math_ops) with keep_dims is deprecated and will be removed in a future version.
Instructions for updating:
keep_dims is deprecated, use keepdims instead
INFO:tensorflow:Scale of 0 disables regularizer.
INFO:tensorflow:Scale of 0 disables regularizer.
WARNING:tensorflow:From /data/tensorflow/models/research/object_detection/core/losses.py:317: softmax_cross_entropy_with_logits (from tensorflow.python.ops.nn_ops) is deprecated and will be removed in a future version.
Instructions for updating:
Future major versions of TensorFlow will allow gradients to flow into the labels input on backprop by default.
See @{tf.nn.softmax_cross_entropy_with_logits_v2}.
我正在使用的配置文件可以找到here。我对eval.py脚本进行了略微编辑,以允许我选择运行脚本的GPU。可以看到here。我正在运行tensorflow-gpu v1.10.0。