RuntimeError

[suhan21]

Traceback (most recent call last):
File "/home/suhan/DLPro/Image-Background-Remover-Python-main/init.py", line 113, in
print(removeBg(imgPath))
File "/home/suhan/DLPro/Image-Background-Remover-Python-main/init.py", line 82, in removeBg
d1, d2, d3, d4, d5, d6, d7 = net(image)

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

Hi , I have install all the necessary packages and i use interpreter as conda. I Got the above error. I can't able to solve. Please, can you help me to solve this.Thanks in advance.

[hassancs91]

what type of images you are testing with?

[suhan21]

I have testing as the same cat image, even i also used python default interpreter / conda(base) interpreter i can't solve runtime error.
Here, i attached the screenshot.

[suhan21]

hassan Please Can you help me to solve this issues

Thanks in advance

[Armando1206]

I have the same problem, can u help us?

[BhaumikThakkar]

hassan Please Can you help me to solve this issues

Thanks in advance

Try on google colab

[njupt99lee]

I have testing as the same cat image, even i also used python default interpreter / conda(base) interpreter i can't solve runtime error. Here, i attached the screenshot.

try this : line 81 image = image.type(torch.FloatTensor) -> image = image.type(torch.cuda.FloatTensor)