Uformer icon indicating copy to clipboard operation
Uformer copied to clipboard

Non-square input

Open Scienceseb opened this issue 3 years ago • 2 comments

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

Scienceseb avatar Oct 12 '21 15:10 Scienceseb

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

ZhendongWang6 avatar Oct 12 '21 15:10 ZhendongWang6

#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) ?

Scienceseb avatar Oct 14 '21 17:10 Scienceseb