-
Notifications
You must be signed in to change notification settings - Fork 45.2k
Open
Labels
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
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
Reactions are currently unavailable
