Is the possible to speed up the inference speed?

[ruanjiyang]

Really cool project! I love it!

Just one question: Is the possible to speed up the inference speed? or any suggestion? thanks!

[qianqianwang68]

Yes, I think that would be possible. One way would be to reduce the size of the feature extraction and image synthesis network. In addition, I haven't thoroughly profiled the inference time; it is possible that the code is not efficient and can be improved. Another way to speed up the inference without changing a lot is to remove the whole feature rendering part and directly render the color images only. See here.

3d-moments/core/renderer.py

Line 167 in e6903f2

direct_rgb = direct_rgb_out[..., :3].permute(0, 3, 1, 2)

But this could also give you a performance drop.