4

I am trying to train the CenterNet HourGlass104 512x512 model for object detection with:

python model_main_tf2.py --model_dir=models/centernet_hg104_512x512_coco17_tpu-8 --pipeline_config_path=models/centernet_hg104_512x512_coco17_tpu-8/pipeline.config

When starting training the error:

AssertionError: Some Python objects were not bound to checkpointed values, likely due to changes in the Python program: [SyncOnReadVariable:{
  0: <tf.Variable 'center_net_hourglass_feature_extractor/hourglass_network/encoder_decoder_block_5/encoder_decoder_block_6/encoder_decoder_block_7/encoder_decoder_block_8/encoder_decoder_block_9/residual_block_60/convolutional_block_62/batchnorm/moving_mean:0' shape=(384,) dtype=float32, numpy=
array([

is thrown.

I have found this Thread, but

fine_tune_checkpoint_type: "detection"

was already set.

Does anyone know what I am missing?

Edit (18/02/21):

I changed fine_tune_checkpoint_type to "fine_tune". With this I can start the training, but I get no output regarding the quality of the results. Checkpoint files are generated.

A-wels
  • 77
  • 1
  • 8

0 Answers0