Skip to content

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

@lacls

Description

@lacls

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

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions