Training code and larger model

baegwangbin / DSINE

[CVPR 2024 Oral] Rethinking Inductive Biases for Surface Normal Estimation

https://baegwangbin.github.io/DSINE/

Other

723 stars 28 forks source link

Training code and larger model #3

Closed AIBluefisher closed 7 months ago

AIBluefisher commented 8 months ago

Hi, @baegwangbin

Thanks for your awesome work! I would like to know will the training code be released and if will you train the model on a larger dataset, e.g. Omnidata v2.

baegwangbin commented 7 months ago

Hi, sorry for the delayed response.

The training code will be released this month (Apr 2024). I currently have a limited compute (one 4090 GPU), so I encourage others to train the model on larger datasets, once the training code is released.