search

ChrisChen1023 / HINT

HINT: High-quality INpainting Transformer with Enhanced Attention and Mask-aware Encoding
MIT License
30 stars 4 forks source link

ValueError: the input array must have size 3 along "channel_axis", got (256,256) #21

Open LQ971225 opened 4 months ago

LQ971225 commented 4 months ago

很感谢你伟大的作品，我想寻求你的帮助，我的数据集是自己的，在test的时候，我将config.yml里面的MSAK:6,运行程序时出现这样的错误，我应该怎样修改呢，迫切需要您的帮助，谢谢！ 2024-07-18 21-13-32屏幕截图

ChrisChen1023 commented 4 months ago

很感谢你伟大的作品，我想寻求你的帮助，我的数据集是自己的，在test的时候，我将config.yml里面的MSAK:6,运行程序时出现这样的错误，我应该怎样修改呢，迫切需要您的帮助，谢谢！ 2024-07-18 21-13-32屏幕截图

Hi there,

Thanks for your interests for our work. It seems like you are inputting a barney image instead of RGB image. In this case you may can copy the this channel 3 times and stack them to make a (3, 256,256) image, then the dataloader should be working. Another way to solve that you use opencv or other tools to transfer this image from gray image to RGB image. Then input it. Hope these can help.

LQ971225 commented 4 months ago

谢谢你如此及时的回复，我已经在尝试了，非常感谢！