search

rstrudel / segmenter

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation
MIT License
846 stars 176 forks source link

Inference Error #10

Closed Elfilali-Taoufiq closed 3 years ago

Elfilali-Taoufiq commented 3 years ago

hello, and thanksn for this great work,

I am trying to do inference from the pretrained model, i chose the first model : Seg-T-Mask/16 | 38.1 / 38.8 | 7M | 512x512

then, i want to test for some images with *512512** resolution, but i got this error

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

Thanks

woctezuma commented 3 years ago

Ditto. https://github.com/rstrudel/segmenter/issues/11#issuecomment-876396362

rstrudel commented 3 years ago

Hi @Elfilali-Taoufiq , I pushed a fix, cf #11 . It should be good now.