vita-group / nerf-sos Goto Github PK

[ICLR2023] "NeRF-SOS: Any-View Self-supervised Object Segmentation from Complex Real-World Scenes", Zhiwen Fan, Peihao Wang, Xinyu Gong, Yifan Jiang, Dejia Xu, Zhangyang Wang

License: MIT License

Python 95.67% Jupyter Notebook 2.55% Shell 1.78%

nerf segmentation self-supervised-learning

nerf-sos's Issues

train custom dataset

Hi!
Thanks for your amazing work. So I want to train it on custom dataset. Is it possible to generate dataset without segments masks, train and get segmentation masks? If masks are necessary how many of them are required ? I have 144 images of dataset + reconstructed poses with colmap.

Hi, im interested in your work, but how to get the segments folder？ Thanks!

tanks and temples Dataset required

I downloaded the dataset from the official website but it does not match the load file you provided. Can you provide a link to the dataset for tanks and samples?

UnboundLocalError: local variable "cls_" referenced before assignment

Congratulations on your good work. I am trying to train NeRF-SOS by using your dataset. However, I am getting an error message;

UnboundLocalError: local variable "cls_" referenced before the assignment.

I am kindly asking for possible causes of this.

Training

Hi, I just try to train with other scenes in llff datasets. How to make a dataset like flower as I noticed that there is a masks and segments subfolder under the flower folder.

Recommend Projects