search

hieubkset / pytorch-image-deblurring

Multi-scale network for image deblurring
20 stars 10 forks source link

Getting an error with real-life images #7

Open EricLe-dev opened 3 years ago

EricLe-dev commented 3 years ago

Dear Hieubkset,

Cảm ơn bạn rất nhiều vì đã chia sẻ. Mình đã follow kĩ tutorial, ở phần demo.py, mình đưa thử vào đó một ảnh bất kì ở kích thước 709x585 và bị lỗi: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 584 and 585 in dimension 2 at C:/w/1/s/windows/pytorch/aten/src\THC/generic/THCTensorMath.cu:71

Theo traceback thì lỗi xuất hiện ở dòng này: output_l1, _, _ = my_model((input_b1, input_b2, input_b3))

Nguyên nhân nằm ở dòng này: output_l1 = self.scale2_net(torch.cat((input_b1, output_l2_up), 1))

Bạn có thể chỉ cho mình nguyên nhân và cách khắc phục được không? Cảm ơn bạn nhiều!

EricLe-dev commented 3 years ago

@hieubkset ngoài ra, mong bạn chỉ cho mình cách áp dụng code của bạn trên grayscale images. Mình đang có một dataset gồm các ảnh grayscale và rất nóng lòng được test thử. Cảm ơn bạn.