Uformer
Uformer copied to clipboard
Non-square input
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
Hi, you can refer to #15 and #13 .
#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) ?