naver / dust3r

DUSt3R: Geometric 3D Vision Made Easy
https://dust3r.europe.naverlabs.com/
Other
4.65k stars 515 forks source link

make logging less verbose #63

Closed hturki closed 3 months ago

hturki commented 3 months ago

Thanks for the great work! Here are some minor proposed changes to make the logging a bit less verbose (that I could desirable when using dust3r inside of larger offline training pipelines). I also tweaked compute_global_alignment so that it doesn't use autocast (within which it seems to currently fail) and actually returns the global alignment loss.

As a feature request, it would be nice if dust3r could be packaged as something installable via pip (ie: either pip install duster or pip install git+https://github.com/naver/dust3r.git)