ModuleNotFoundError: No module named 'bvh_distance_queries_cuda' about icon HOT 7 CLOSED

@1kaiser

from icon.

Hi @1kaiser, thanks for your suggestions. I have updated the fetch scripts for SMPL downloading.

But please remember NOT to include any personal info in Github's issues, i.e., username, email, and password (I have removed them for you) due to the privacy concern.

Anyone who wanna use these models MUST register at these sites and sign the required agreement.

from icon.

Have you installed or compiled all the required packages in installation?

This package comes from cuda src, the error should be fixed after compiling the bvh-distance-queries.

Please let me know if it helps.

from icon.

I missed/overlooked that installation page. I am sorry. Thank you so much for your help. :) :) :)

from icon.

colab version anytime ??

from icon.

colab version anytime ??

soon

from icon.

!git clone https://github.com/YuliangXiu/ICON.git
!cd /content/ICON

!pip install -r /content/ICON/requirements.txt

# install rembg
!GPU=1 pip install rembg --upgrade

# install neural_renderer
#cuda >= 11.0
!git clone https://github.com/adambielski/neural_renderer.git
!cd neural_renderer && python setup.py install
# cuda < 11.0
!pip install neural_renderer_pytorch

# install smplx (modified by @Yuliang Xiu)
!git clone https://github.com/YuliangXiu/smplx.git
!cd smplx && python setup.py install

# install bvh-distance-queries
!cd /content/ICON/lib/bvh-distance-queries && git clone https://github.com/NVIDIA/cuda-samples.git
!cd /content/ICON/lib/bvh-distance-queries && export CUDA_SAMPLES_INC='/content/ICON/lib/bvh-distance-queries/cuda-samples/Common' && python setup.py install

and to download the required files we had to login from lots of sites
since everyone would be directly running the code in colab

little backend file downloading and resarranging to structyure is also required

Pre-trained models and extra data

!mkdir -p /content/data/pymaf_data/a/aa
!wget http://visiondata.cis.upenn.edu/spin/data.tar.gz -O /content/data/pymaf_data/data.tar.gz && tar xvf /content/data/pymaf_data/data.tar.gz -C /content/data/pymaf_data/a/aa
!mv /content/data/pymaf_data/a/aa/data/* /content/data/pymaf_data
!rm -r /content/data/pymaf_data/a /content/data/pymaf_data/data.tar.gz

USERNAME =
PASSWD =

SMPL (10 shape PCs)
!wget --post-data "username=USERNAME&password=PASSWD" 'https://download.is.tue.mpg.de/download.php?domain=smpl&sfile=SMPL_python_v.1.0.0.zip&resume=1' -O '/content/data/SMPL.zip' --no-check-certificate --continue
!mkdir -p /content/data/a/aa
!unzip /content/data/SMPL.zip -d /content/data/a/aa
!mv /content/data/a/aa/smpl/models/basicModel_f_lbs_10_207_0_v1.0.0.pkl /content/data/smpl_related/models/smpl/SMPL_FEMALE.pkl
!mv /content/data/a/aa/smpl/models/basicmodel_m_lbs_10_207_0_v1.0.0.pkl /content/data/smpl_related/models/smpl/SMPL_MALE.pkl
!rm -r /content/data/SMPL.zip /content/data/a

SMPLIFY_CODE_V2.ZIP
!wget --post-data "username=USERNAME&password=PASSWD" 'https://download.is.tue.mpg.de/download.php?domain=smplify&sfile=mpips_smplify_public_v2.zip&resume=1' -O '/content/data/SMPL_neutral.zip' --no-check-certificate --continue
!mkdir -p /content/data/a/aa
!unzip /content/data/SMPL_neutral.zip -d /content/data/a/aa
!mv /content/data/a/aa/data/smplify_public/code/models/basicModel_neutral_lbs_10_207_0_v1.0.0.pkl /content/data/smpl_related/models/smpl/SMPL_NEUTRAL.pkl
!rm -r /content/data/a /content/data/SMPL_neutral.zip

Download [ICON]
!wget --post-data "username=USERNAME&password=PASSWD" 'https://download.is.tue.mpg.de/download.php?domain=icon&sfile=icon_data.zip&resume=1' -O '/content/data/icon_data.zip' --no-check-certificate --continue
!mkdir -p /content/data/a/aa
!unzip /content/data/icon_data.zip -d /content/data/a/aa
!mv /content/data/a/aa/smpl_data /content/data/smpl_related
!rm -r /content/data/a /content/data/icon_data.zip

from icon.