Shape Mismatch When Running Demo on Images

[MD-Student]

I am trying to run the demo code on images. After following the readme guide to set up python environment, I encountered shape mismatch errors as below. The only change I did to the code is that I changed the pyopengl_platform from egl to osmesa because egl couldn't work on my Ubuntu system. I am not sure whether the following error comes from this change. I would be very happy if anyone could offer any help. Thanks sincerely.

ValueError: operands could not be broadcast together with shapes (225,300,3) (225,300,0)