tensorflow / tensorflow/models

[FasterRCNN] Checkpoint is expected to be an object-based checkpoint_Object detection.

Open
#9,278 14 comments 4 reactions 3 assignees View on GitHub

@pkulzc is already working on this.

Since Oct 9, 2020.

models:research:odapi type:bug
Dominant language
Python
Stars
77.7k
Forks
44.8k
PR merge metrics
No merged PRs in 30d

Description

Prerequisites

  • I am using the model faster_rcnn_resnet101_v1_800x1333_coco17_gpu.

https://github.com/tensorflow/models/tree/master/research/object_detection

1. Describe the bug

My running on the terminal

python object_detection/model_main_tf2.py \
    --pipeline_config_path=/home/MaskRCNN_DeepFashion2/models/research/object_detection/configs/tf2/faster_rcnn_resnet101_v1_800x1333_coco17_gpu-8.config \
    --model_dir=home/MaskRCNN_DeepFashion2/Mask_RCNN/CP \
    --num_train_steps=10000 \
    --sample_1_of_n_eval_examples=1 \
    --alsologtostderr

The error I got is below

  File "object_detection/model_main_tf2.py", line 113, in <module>
    tf.compat.v1.app.run()
  File "/home/anaconda3/lib/python3.7/site-packages/tensorflow/python/platform/app.py", line 40, in run
    _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
  File "/home/anaconda3/lib/python3.7/site-packages/absl/app.py", line 299, in run
    _run_main(main, args)
  File "/home/anaconda3/lib/python3.7/site-packages/absl/app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "object_detection/model_main_tf2.py", line 110, in main
    record_summaries=FLAGS.record_summaries)
  File "/home/anaconda3/lib/python3.7/site-packages/object_detection-0.1-py3.7.egg/object_detection/model_lib_v2.py", line 569, in train_loop
    unpad_groundtruth_tensors)
  File "/home/anaconda3/lib/python3.7/site-packages/object_detection-0.1-py3.7.egg/object_detection/model_lib_v2.py", line 346, in load_fine_tune_checkpoint
    raise IOError('Checkpoint is expected to be an object-based checkpoint.')
OSError: Checkpoint is expected to be an object-based checkpoint.

Here is my config file with ONLY Fine-tune part

 }
  gradient_clipping_by_norm: 10.0
  fine_tune_checkpoint: "/home/MaskRCNN_DeepFashion2/pre_trained_models/faster_rcnn_resnet101_coco_2018_01_28/model.ckpt"
  fine_tune_checkpoint_type: "detection"

The folder of pre_trained_models consists of

Screen Shot 2020-09-22 at 07 12 21

Thanks you guys, again!

6. System information

  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Linux
  • TensorFlow installed from (source or binary):
  • TensorFlow version (use command below): v2.2.0-rc4-8-g2b96f3662b || 2.2.0
  • Python version: 3.7.6
  • CUDA/cuDNN version: 10.2
  • GPU model and memory: Quardo RTX 8000 48601Mib

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.