median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
1.94k stars 181 forks source link

Image size of NYUv2 dataset should be 3*288*384 #61

Closed Xjh0327 closed 10 months ago

Xjh0327 commented 10 months ago

The README.md in ./examples/nyu says it's 3*284*384 while it should be 3*288*384. :)

Baijiong-Lin commented 10 months ago

The NYUv2 dataset is pre-processed by mtan.

Xjh0327 commented 10 months ago

Yes, I've seen this. But in this paper it also says the image size is 3*288*384. Setting --img_size 3 284 384 will raise bug. Just a small typo in README.md. Thank you!

Baijiong-Lin commented 10 months ago

Sorry for my misunderstanding. Thanks for pointing out this typo.