Hi Piao, thanks for your awesome works. Currently I find that there is only traning code released. I wonder if it is possible to release the pretrained checkpoint for some testing and evaluation. Or is it possible to send you several images to observe the performances? Looking forward to your reply.

in /op/rasterize.py
    when i try:

    if use_cuda:
	v = v.cuda(1)
	f = f.cuda(1)
	t = t.cuda(1)
o = rasterize(v, t, f, 5)
   the process keep running, and got nothing from the print() function;
   if try it in torch.distributed, got errors like:
   CUDA error: an illegal memory access was encountered.

Hi,

Thank you for sharing the code of the great work!

I have been adopt the code of the generator to my own project, and I am concerned with the following line of code: https://github.com/WestlyPark/StyleRenderer/blob/a1a093a030716ceb24b46e2fd48d2580f7429463/model.py#L216

I feel this line is unnecessary as the to_rgb layers are already initialized in https://github.com/WestlyPark/StyleRenderer/blob/a1a093a030716ceb24b46e2fd48d2580f7429463/model.py#L192 If I understand it corretly, the current version of code will double the amount of to_rgb layers, but only the first half are used in the forward process (the generation results are still good though).

Best regards

Hi When running th code, the ['v '], ['tex'], ['tri '] attributes of model are not found in BFM.mat. the BFM.mat (01_MorphableModel.mat) is downloaded from the BFM2009 webside: https://faces.dmi.unibas.ch/bfm/main.php?nav=1-0&id=basel_face_model
As we all know， 01_MorphableModel.mat does not have th key of ['v '], ['tex']. Can you share your BFM.mat，or the code to get it?

Thank you for your great work. But when I run this commond "python uitls_face.py --lmk dlib --bfm BFM.mat --output OUTPUT_PATH DATASET_PATH", I encounter some issues like below: Traceback (most recent call last): File "H:/stylerender/StyleRenderer/utils_face.py", line 522, in from op import rasterize File "H:\stylerender\StyleRenderer\op_init_.py", line 3, in from .rasterize import rasterize File "H:\stylerender\StyleRenderer\op\rasterize.py", line 15, in os.path.join(module_path, 'rasterize.cu')
File "C:\Users\Administrator\anaconda3\envs\styleflow\lib\site-packages\torch\utils\cpp_extension.py", line 898, in load is_python_module) File "C:\Users\Administrator\anaconda3\envs\styleflow\lib\site-packages\torch\utils\cpp_extension.py", line 1097, in _jit_compile return _import_module_from_library(name, build_directory, is_python_module) File "C:\Users\Administrator\anaconda3\envs\styleflow\lib\site-packages\torch\utils\cpp_extension.py", line 1422, in _import_module_from_library file, path, description = imp.find_module(module_name, [path]) File "C:\Users\Administrator\anaconda3\envs\styleflow\lib\imp.py", line 302, in find_module raise ImportError(_ERR_MSG.format(name), name=name) ImportError: No module named 'rasterize'

Whether I need to complie rasterize.cpp and rasterize.cu? Or mayebe the CUDA and pytorch version are wrong?