HSI_SSFTT
HSI_SSFTT copied to clipboard
zgr6010
L. Sun, G. Zhao, Y. Zheng, and Z. Wu. "Spectral–Spatial Feature Tokenization Transformer for Hyperspectral Image Classification," in IEEE TGRS, 2022.
SSFTTnet.py下SSFTTnet类下的forward： “wa = rearrange(self.token_wA, 'b h w -> b w h') # Transpose A = torch.einsum('bij,bjk->bik', x, wa)” wa是torch.Size([1, 64, 4]) x是torch.Size([64, 81, 64]) 这块bij bjk里面的b不一样
作者您好！在绘制图片的时候，遇到了如下问题，希望您有空能够帮忙解答一下！ --------------------------------------------------------------------------- IndexError Traceback (most recent call last) /tmp/ipykernel_25/2069759544.py in 611 print(y.shape[0]) 612 print(y.shape[1]) --> 613 get_cls_map(net, device, all_data_loader, y_all) /tmp/ipykernel_25/2069759544.py in get_cls_map(net, device, all_data_loader, y) 113 114 y_pred, y_new...
