search

google-research / 3d-moments

Code for CVPR 2022 paper '3D Moments from Near-Duplicate Photos'
Apache License 2.0
245 stars 27 forks source link

Is the possible to speed up the inference speed? #4

Open ruanjiyang opened 2 years ago

ruanjiyang commented 2 years ago

Really cool project! I love it!

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

qianqianwang68 commented 2 years ago

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. https://github.com/google-research/3d-moments/blob/e6903f29736a1369daf75642eacb7a8836539e77/core/renderer.py#L167 But this could also give you a performance drop.