caiyuanhao1998 / MST-plus-plus

"MST++: Multi-stage Spectral-wise Transformer for Efficient Spectral Reconstruction" (CVPRW 2022) & (Winner of NTIRE 2022 Spectral Recovery Challenge) and a toolbox for spectral reconstruction

Home Page:https://arxiv.org/abs/2204.07908

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

数据集处理疑问

dongruoyu1 opened this issue · comments

image
您好,在阅读您的代码时,发现了对训练集加载的时候,有这样两段代码。不是很理解他的用意,我计算得self.patch_per_img为2205,那么每一张图片的索引为什么会是idx//self.patch_per_img?而idx从0-900,那岂不是img_idx都是0了。
img_idx, patch_idx = idx//self.patch_per_img, idx%self.patch_per_img
h_idx, w_idx = patch_idx//self.patch_per_line, patch_idx%self.patch_per_line

每张图片切成多个patch,
训练的instance是一个patch

如果觉得有帮助的话,帮我们点点 star 支持一下