search

caiyuanhao1998 / MST

A toolbox for spectral compressive imaging reconstruction including MST (CVPR 2022), CST (ECCV 2022), DAUHST (NeurIPS 2022), BiSCI (NeurIPS 2023), HDNet (CVPR 2022), MST++ (CVPRW 2022), etc.
MIT License
637 stars 71 forks source link

self.HSI #17

Closed Sebastian970107 closed 1 year ago

Sebastian970107 commented 1 year ago

在dataset中有写到如果是test模式的话为 hsi = self.HSI[:, :, :, index1] 请问这里的HSI是什么数据集呢，代码中似乎没有交代。

caiyuanhao1998 commented 1 year ago

你说的这个代码的位置在哪？

Sebastian970107 commented 1 year ago

real/train_code/dataset.py 除了cave和kaist之外似乎应该还有一个数据集

caiyuanhao1998 commented 1 year ago

对的，real进行test的时候，用到了一个真实的数据集，是由TSA-Net这篇文章搜集的。我们也有在data_preparation里面提供下载链接