search

txchen-USTC / MiM-ISTD

Official pytorch code of our paper "MiM-ISTD: Mamba-in-Mamba for Efficient Infrared Small Target Detection"
MIT License
102 stars 10 forks source link

val model error: backbone #5

Open SeaLu opened 5 months ago

SeaLu commented 5 months ago

When I input 640*640 images to train the backbone of the model, the error is
erroe

SeaLu commented 5 months ago

When I val the model, the error is : mid_out=outer_tokens.reshape(b,int(math.sqrt(l)),int(math.sqrt(l)),m).permute(0,3,1,2) RuntimeError: shape '[32, 75, 75, 16]' is invalid for input of size 2924544

avenger123456 commented 5 months ago

Hi，I can help you with that, would you mind telling me your email?

DuHao55 commented 5 months ago

I input 512×512 images into the network and no errors are reported.I think reporting this error has something to do with the size of the image. @SeaLu