ZhendongWang6 / Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

MIT License

789 stars 116 forks source link

Non-square input #23

Open Scienceseb opened 2 years ago

Scienceseb commented 2 years ago

Hi your code curently dont support non-square entry, even if I change img_size for img_size_H and img_size_W, there is still an error because the input is not a square. This is a very big limitation of your code right now..

Kind regards, Sébastien de Blois

ZhendongWang6 commented 2 years ago

Hi, you can refer to #15 and #13 .

Scienceseb commented 2 years ago

15 and #13 are for testing mode, not training mode... is your solution to my problem to use the code of expand2square(timg,factor=X) ?