tensorflow对象检测api评估分割故障

无论我使用object_detection/legacy/eval.py脚本还是object_detection/model_main.py,无论何时开始评估,我都存在细分错误。 我试图重新安装tensorflow,protobuf编译器并重新安装tensorflow / modules对象检测api的所有依赖项,但没有帮助。我正在使用python 3。

Tensorflow版本:1.15.0

我给eval.py打电话的方式:

python object_detection/legacy/eval.py \
    --logtostderr \
    --eval_dir=${TRAIN_DIR} \
    --checkpoint_dir=${TRAIN_DIR} \
    --pipeline_config_path=${PIpelINE_CONFIG_PATH}

这是运行脚本时的日志段

I1107 17:50:24.818984 140633596692288 saver.py:1284] Restoring parameters from /home/ihahanov/Projects/rooftops/resources/models/mask_rcnn_inception_v2_coco_2018_01_28/train/model.ckpt-29722
2019-11-07 17:50:27.309148: W tensorflow/core/framework/cpu_allocator_impl.cc:81] Allocation of 92160000 exceeds 10% of system memory.
Fatal Python error: Segmentation fault

Thread 0x00007fe6c4ff9700 (most recent call first):
  File "/usr/lib/python3.6/threading.py",line 295 in wait
  File "/usr/lib/python3.6/threading.py",line 551 in wait
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/training/coordinator.py",line 311 in wait_for_stop
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/training/queue_runner_impl.py",line 293 in _close_on_stop
  File "/usr/lib/python3.6/threading.py",line 864 in run
  File "/usr/lib/python3.6/threading.py",line 916 in _bootstrap_inner
  File "/usr/lib/python3.6/threading.py",line 884 in _bootstrap

Thread 0x00007fe6bbfff700 (most recent call first):
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1443 in _call_tf_sessionrun
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1287 in _single_operation_run
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/training/queue_runner_impl.py",line 257 in _run
  File "/usr/lib/python3.6/threading.py",line 884 in _bootstrap

Thread 0x00007fe7cf930740 (most recent call first):
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1350 in _run_fn
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1365 in _do_call
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1359 in _do_run
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 1180 in _run
  File "/home/ihahanov/Projects/rooftops/venv/lib/python3.6/site-packages/tensorflow_core/python/client/session.py",line 956 in run
  File "/home/ihahanov/Projects/rooftops/models/research/object_detection/legacy/evaluator.py",line 234 in _process_batch
  File "/home/ihahanov/Projects/rooftops/models/research/object_detection/eval_util.py",line 346 in _run_checkpoint_once
  File "/home/ihahanov/Projects/rooftops/models/research/o/home/ihahanov/Projects/rooftops/shell/train_model.sh: line 26: 13874 Segmentation fault      (core dumped) python3 eval.py --logtostderr --eval_dir=${TRAIN_DIR} --checkpoint_dir=${TRAIN_DIR} --pipeline_config_path=${PIpelINE_CONFIG_PATH}
pl3000 回答:tensorflow对象检测api评估分割故障

即使在注释部分中也提供了解决方案(答案部分),还是为了社区的利益。

由于模型中涉及的计算非常占用内存,因此在GPU(而不是CPU)中执行模型可以解决该错误。

本文链接:https://www.f2er.com/3143347.html

大家都在问