magicleap / Atlas

Atlas: End-to-End 3D Scene Reconstruction from Posed Images
Apache License 2.0
1.82k stars 221 forks source link

Pretrained plane model #8

Open GainiK opened 4 years ago

GainiK commented 4 years ago

There is a retrained model with semantic segmentation head. Can you please also upload the plain one?

zmurez-ml commented 4 years ago

We do not plan to release the plain one. The results are very similar to the one with semantic segmentation. You can simply ignore the semseg output (there is very little additional compute required to run the additional head). If you really want the semseg head gone, you can alter the config file and remove the respective weights from the checkpoint, or retrain a plain one from scratch.