自己的数据集训练CosinGAN,报错IndexError:too many indices for array:array is 2-dimensional, but 3 were indexed

在尝试使用自己的SAR图像数据集(800*800,灰度图,三通道)训练CosinGAN时,遇到IndexError。原本论文中彩色图像训练正常,但SAR图像导致错误。问题在于img.imread无法正确读取SAR图像的三通道。解决方案是改用OpenCV来读取图像,确保正确处理图像通道。
摘要由CSDN通过智能技术生成

感谢室友伟哥~

用自己的SAR图像数据集训练CosinGAN, 图像尺寸为800*800,灰度图,三通道
用论文提供的彩色图像训练时是好着的,但换成自己的SAR图像就报错:

IndexError:too many indices for array:array is 2-dimensional, but 3 were indexed

本以为SAR图像为单通道的就一直想改成三通道的,结果还是不行,玄学:
functions.py中读取图像的函数

def read_image(opt):
    x = img
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值