InvalidArgumentError when restoring the checkpoint about pixel2mesh

[chongyangma]

@M4gicT0 After change the code, have you rerun python setup.py install?

[chongyangma]

I ran into the same error. It looks like changing the following two lines in layers.py:

x2 = tf.ceil(x)

and

y2 = tf.ceil(y)

into

x2 = tf.minimum(tf.ceil(x), tf.cast(tf.shape(img_feat)[0], tf.float32) - 1)

and

y2 = tf.minimum(tf.ceil(y), tf.cast(tf.shape(img_feat)[1], tf.float32) - 1)

will bypass the error. Not sure if this is the correct solution.

[DubiousCactus]

@chongyangma Didn't do that, it works now. Thanks !

[DubiousCactus]

Unfortunately it didn't fix anything for me...

[DubiousCactus]

@nywang16 Is that a proper fix ? It should work out of the box.

[hinnBU]

Ran into this problem currently. We found that the problem is about the miscalculated indices.
Layer dimensions (see api.py or model.py):

56x56 --> indices[0...55]
28x28 --> indices[0..27]
14x14 --> indices[0..13]
7x7 --> indices[0..6]

Lines 248-253 of layers.py:

		h = tf.minimum(tf.maximum(h, 0), 223)
		w = tf.minimum(tf.maximum(w, 0), 223)

		x = h/(224.0/56)
		y = w/(224.0/56)
		out1 = project(self.img_feat[0], x, y, 64)

Calculation:

• h and w can be values from 0 to 223
• 223 / (224 / 56) = 55.75
• tf.ceil(55.75) = 56

So we try to index our range of 0..55 mentioned earlier with 56 and this is the reason for the error.

Fix:

• 223 / (223 / 55) = 55
• tf.ceil(55.0) = 55

This has to be done for every x,y calculation before project() call.

Funny thing is this error only occurs if TF is running on CPU.
While running on GPU the call returns 0:

https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/gather_nd

Retraining is needed after the fix.