NonCuboidRoom icon indicating copy to clipboard operation
NonCuboidRoom copied to clipboard

Inference on custom data

Open Hui-Yao opened this issue 3 years ago • 2 comments

I am doing some inference using the Structure3D_pretrained.pt model which is downloaded from this repo, and the custom image is come from the InteriorNet dataset, which is introduced by the KuJiaLe, too. The image size is (480, 640), and the camera intrinsic is [[600, 0, 320], [0, 600, 240], [0, 0, 1].

Part of the result output by the model seems to be reasonable, but other result is hard to accept. So should i change the intrinsic matrix ? or how should i modify the hyperparameter setting ? Any suggestion will be grateful.

In the following pictures, red is the GT edge and green is predicted by the model: comparison_1509_1602_3FO4II8OLSP6_Dining_room_6 comparison_1201_1301_3FO4IGB214OC_Dining_room_17 comparison_807_909_3FO4IDX9JIBX_Bathroom_2 comparison_910_1011_3FO4IEQVRLOT_Kids_room_5 comparison_2607_2702_3FO4ILIXGI9L_Dining_room_13

Hui-Yao avatar Nov 24 '21 13:11 Hui-Yao

Which one is the model prediction? The green? If yes what is the red instead?

marcomiglionico94 avatar Nov 24 '21 15:11 marcomiglionico94

Which one is the model prediction? The green? If yes what is the red instead?

The red one is the Ground Truth edge, and the green one is predicted by the model.

Hui-Yao avatar Nov 25 '21 02:11 Hui-Yao