search

Reagan1311 / DABNet

Depth-wise Asymmetric Bottleneck for Real-time Semantic Segmentation (BMVC2019)
https://github.com/Reagan1311/DABNet
MIT License
140 stars 36 forks source link

Prediction on resized image #29

Closed Zeeshan313 closed 4 years ago

Zeeshan313 commented 4 years ago

Hello, i want to make prediction on resized image. i means that i want to rescale image from (1024x2048) to (512X1024) to increase fps. but when i use predict function to do prediction. it throw me error. "self.padding, self.dilation, self.groups) RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same."

can you please help me to solve this problem.

Reagan1311 commented 4 years ago

Hi, there is a cityscapes.py in the dataset folder, where you can find a f_scale in the function CityscapesValDataSet, you can change this to rescale the image.