search

ybkscht / EfficientPose

Other
247 stars 68 forks source link

Shape mismatch #32

Closed nickhward closed 3 years ago

nickhward commented 3 years ago

When I try to run the command: python inference.py I get the following error.

ValueError: Layer #480 (named "class_net/class-predict"), weight <tf.Variable 'class_net/class_net/class-predict/pointwise_kernel:0' shape=(1, 1, 64, 72) dtype=float32> has shape (1, 1, 64, 72), but the saved weight has shape (1, 1, 64, 9). Does anyone know how to fix this issue?

nickhward commented 3 years ago

Dumb mistake....

In inference.py I just had to change the class_to_name to the driller class because I trained with linemod.