I encountered an issue while trying to reproduce the method described in your repository.

[maoliangjie]

Specifically, I received the following error:

Traceback (most recent call last):
File "./get_bfm_coeff.py", line 118, in
main(0, opt, opt.img_folder)
File "./get_bfm_coeff.py", line 86, in main
model.test() # run inference
File "/workspace/AT3D/preprocess/get_bfm_mat/Deep3DFaceRecon_pytorch/models/base_model.py", line 162, in test
self.forward()
File "/workspace/AT3D/preprocess/get_bfm_mat/Deep3DFaceRecon_pytorch/models/facerecon_model.py", line 140, in forward
self.pred_vertex, self.facemodel.face_buf, feat=self.pred_color)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/workspace/AT3D/preprocess/get_bfm_mat/Deep3DFaceRecon_pytorch/util/nvdiffrast.py", line 64, in forward
self.ctx = dr.RasterizeCudaContext(device=device)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/nvdiffrast/torch/ops.py", line 177, in init
self.cpp_wrapper = _get_plugin().RasterizeCRStateWrapper(cuda_device_idx)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/nvdiffrast/torch/ops.py", line 118, in _get_plugin
torch.utils.cpp_extension.load(name=plugin_name, sources=source_paths, extra_cflags=opts, extra_cuda_cflags=opts+['-lineinfo'], extra_ldflags=ldflags, with_cuda=True, verbose=False)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 1091, in load
keep_intermediates=keep_intermediates)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 1317, in _jit_compile
return _import_module_from_library(name, build_directory, is_python_module)
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 1699, in _import_module_from_library
file, path, description = imp.find_module(module_name, [path])
File "/root/anaconda3/envs/deep3d_pytorch/lib/python3.7/imp.py", line 296, in find_module
raise ImportError(_ERR_MSG.format(name), name=name)
ImportError: No module named 'nvdiffrast_plugin'

[elenacliu]

Sorry for the late reply.

ImportError: No module named 'nvdiffrast_plugin'

It seems that you didn't install the nvdiffrast plugin and it is necessary for the rendering process.

You can refer to this code:

AT3D/run_preprocess.sh

Lines 14 to 20 in f0233c1

	if [ ! -d "./nvdiffrast/" ]; then
	git clone https://github.com/NVlabs/nvdiffrast
	fi
	cd ./nvdiffrast
	git checkout fad71a4
	pip install .