Skip to content
Snippets Groups Projects
Commit 690e0839 authored by Thomas Müller's avatar Thomas Müller
Browse files

`colmap2nerf.py --mask_categories`: error if PyTorch is not installed

parent 1b81da33
No related branches found
No related tags found
No related merge requests found
......@@ -396,6 +396,12 @@ if __name__ == "__main__":
try:
import detectron2
except ModuleNotFoundError:
try:
import torch
except ModuleNotFoundError:
print("PyTorch is not installed. For automatic masking, install PyTorch from https://pytorch.org/")
sys.exit(1)
input("Detectron2 is not installed. Press enter to install it.")
import subprocess
package = 'git+https://github.com/facebookresearch/detectron2.git'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment