Bad shape for unary energy

[anarnold2]

Hi,

I got the same issue now but somehow my code worked the day before. Did you do an update on your library?

This is the error, I got.

 25         U = unary_from_softmax(processed_prediction)
 26         U = np.ascontiguousarray(U) # necessary since the library pydensecrf is using a cython wrapper DEBUg

---> 27 dcrf_obj.setUnaryEnergy(U) 28 29 #pairwise potentials: color-independent -> only on feature location

pydensecrf/densecrf.pyx in pydensecrf.densecrf.DenseCRF.setUnaryEnergy()

ValueError: Bad shape for unary energy (Need (2, 1152), got (2, 147456))

[anarnold2]

Ok my bad, I had an issue in my code!