cnhaox/NeRF-HuGS

segment_anything module

Closed this issue · 2 comments

from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor
from segment_anything.utils.amg import calculate_stability_score

It seems that there is no segment-anything module provided:ModuleNotFoundError: No module named 'segment_anything

I am training the photoourism dataset, and the generate-static_mask.py prompt shows that there is no segmentation module

I am training the photoourism dataset, and the generate-static_mask.py prompt shows that there is no segmentation module

Hi there! This looks like the segment-anything module is not installed in your conda environment. I've just updated the README.md and requirements_torch.txt, and you can follow them to prepare the environment.