search

ywyue / RoomFormer

[CVPR 2023] RoomFormer: Two-level Queries for Single-stage Floorplan Reconstruction
https://ywyue.github.io/RoomFormer/
MIT License
163 stars 22 forks source link

How to visualize the result? #21

Closed K25801 closed 5 months ago

K25801 commented 5 months ago

Hi, thanks for your beautiful work. How can I visualize the results from the processed scenecad data?

K25801 commented 5 months ago

In detail, when I'm testing scenecad datasets, the output I have is this: image image

Seems not to be a well-reconstructed floor plan, is there any suggestions?

ywyue commented 5 months ago

Hi, thanks for your interest and kind words! I am sorry for the late reply.

How can I visualize the results from the processed scenecad data?

You can visualize both the prediction and ground truth by setting --plot_pred , --plot_density, and --plot_gt as True in eval.py.

Seems not to be a well-reconstructed floor plan, is there any suggestions?

I am not sure why you think this is not a well-reconstructed floor plan. I have attached the raw 3D scan and ground truth annotation for this scene below: GT

As you can see, your attached output of the model is quite similar to the ground truth even though the 3D scan is quite noisy. Let me know if you have any further concerns!