search

dragen1860 / TensorFlow-2.x-Tutorials

TensorFlow 2.x version's Tutorials and Examples, including CNN, RNN, GAN, Auto-Encoders, FasterRCNN, GPT, BERT examples, etc. TF 2.0版入门实例代码，实战教程。
6.38k stars 2.24k forks source link

error about fasterRcnn #53

Open ainilian opened 4 years ago

ainilian commented 4 years ago

ValueError Traceback (most recent call last)

in () 5 plt.savefig('image_demo_random.png') 6 ----> 7 model.load_weights('weights/faster_rcnn.h5', by_name=True) ~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/training.py in load_weights(self, filepath, by_name, skip_mismatch) 232 raise ValueError('Load weights is not yet supported with TPUStrategy ' 233 'with steps_per_run greater than 1.') --> 234 return super(Model, self).load_weights(filepath, by_name, skip_mismatch) 235 236 @trackable.no_automatic_dependency_tracking ~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/engine/network.py in load_weights(self, filepath, by_name, skip_mismatch) 1218 if by_name: 1219 hdf5_format.load_weights_from_hdf5_group_by_name( -> 1220 f, self.layers, skip_mismatch=skip_mismatch) 1221 else: 1222 hdf5_format.load_weights_from_hdf5_group(f, self.layers) ~/software/anaconda3/envs/tf2/lib/python3.6/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py in load_weights_from_hdf5_group_by_name(f, layers, skip_mismatch) 775 symbolic_weights[i])) + 776 ', but the saved weight has shape ' + --> 777 str(weight_values[i].shape) + '.') 778 779 else: ValueError: Layer #4 (named "b_box_head"), weight has shape (1024, 21), but the saved weight has shape (1024, 81).