高光谱图像通道编码

目录

现在给高光谱图像加上通道编码

40个通道，loss训练结果：

---

transformer有位置编码，difussion有时间编码

现在给高光谱图像加上通道编码

class SalImgDataset(Dataset):
    def __init__(self,root_path,dtype='val',train_type='torch',transform=None):
        # self.root_dir = root_dir
        # self.image_list = glob.glob(image_dir+'*.png')
        # self.json_list = glob.glob(label_dir+'*.png')
        self.transform = transform
        self.dtype = dtype
        self.train_type = train_type

        self.file_list = ['%s/%s' % (i[0].replace("\\", "/"), j) for i in os.walk(root_path) for j in i[-1] if  j.endswith (('.npy', 'val.xlsx'))]


    def __len__(self):
        return len(self.file_list)*10

    def __getitem__(self,idx):