3DTopia/OpenLRM

Error in exporting mesh and video when applying official examples

Closed this issue · 2 comments

similar as closed issue #1, but I did not find an answer.

when I try the official example exporting videos, the videos are blank
python -m lrm.inferrer --model_name openlrm-base-obj-1.0 --source_image ./assets/sample_input/owl.png --export_video

In the videos I output, it shows a white image with a faintly visible object rotating in the center, and the object is very unclear with the white background. I have tried this on both 2080ti and 3090, also used the openlrm-small-obj-1.0 model and other example images, but the results are similar.

when I try the official example exporting mesh the vertices are empty.

python -m lrm.inferrer --model_name openlrm-base-obj-1.0 --source_image ./assets/sample_input/owl.png --export_mesh

lrm/models/rendering/synthesizer.py", line 189, in forward_points

for k in outs[0].keys()

IndexError: list index out of range

I looked at the result after executing forward_planes in line 165 inferrer.py.
forwardplane

and result after excuting model.synthesizer in line 157 inferrer.py.,in which the images_rgb values for each frame seem to be abnormal.
image_rgb

How can I fix it and perform similar as official example?

Hi Mengjiao,

We haven't encountered such issue in our environment. If you are still seeing this problem, could you plz share your environment versions and other useful information in the log?

Dear authors,
Thank you for the reply and the update! I changed my cuda version to 11.8 and everything works now.
Previous environment cuda11.3, torch 2.2.0
Now environment cuda 11.8, torch 2.2.0