xxlong0 / SparseNeuS

SparseNeuS: Fast Generalizable Neural Surface Reconstruction from Sparse views
MIT License
325 stars 16 forks source link

Confusion about the mask in the general training #22

Open jinnan-chen opened 2 years ago

jinnan-chen commented 2 years ago

Hi, I download the dtu_training.rar with your provided link and do not find folder of Masks. However, the dataloader of dtu_general.py contains such path of mask_filename = os.path.join(self.root_dir, f'Masks/{scan}train/mask{vid:04d}.png'). For the training images in the folder of Rectified, the backgrounds seem to be white. I am not sure whether your code regards the white region as mask instead of mask file for training. Could you please clarify it, thanks!

flamehaze1115 commented 2 years ago

You can download the masks here: https://connecthkuhk-my.sharepoint.com/:u:/g/personal/xxlong_connect_hku_hk/EW_v7RA73HNEquScVtNJ34gB4hYlRfEatW4TOg086F0_Lg?e=3SKiif Training without the masks will be not a problem, just simply set the masks to 1.