pytorch-image-deblurring
pytorch-image-deblurring copied to clipboard
hieubkset
Getting an error with real-life images
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!
@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.