google / lasr

Code for "LASR: Learning Articulated Shape Reconstruction from a Monocular Video". CVPR 2021.
https://lasr-google.github.io/
Apache License 2.0
170 stars 17 forks source link

No module named 'detectron2.config' #8

Open Kana-alt opened 3 years ago

Kana-alt commented 3 years ago

I built my environment with docker. Therefore, I use the following command to get segmentations.

docker run -v $(pwd):/lasr --gpus all lasr bash -c 'cd lasr/detectron2; source activate lasr; python mask.py pika . /detectron2; cd -'

Then I get the following error.

Traceback (most recent call last): File "mask.py", line 23, in from detectron2.config import get_cfg ModuleNotFoundError: No module named 'detectron2.config'

detectron2 is installed with a folder created in the parent directory of preprocess. Is it because I am using docker that I am getting this error?

gengshan-y commented 3 years ago

Yes, the provided docker script would not install and clone detectron2. You may need to modify the docker file to do so. I'll try to push a fix. In the meanwhile, you may use the conda environment to extract segmentations.