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

issue about corners_pad[:len(corners)] = corners #18

Closed Martin-1999 closed 9 months ago

Martin-1999 commented 9 months ago

I got this problem when training on my own dataset

1701153750936 It looks like the length of the corners is 298, but the length of the corners_pad is only 80 Is this caused by my training data? be like: image

thank you very much!

ywyue commented 9 months ago

Thanks for your interest in our work. We assume the maximum number of vertices for each polygon is 40 - that's why the length of the corners_pad [(x1,y1), (x2,y2), ... ] is only 80 (40×2). It seems that your dataset contains a polygon with 298/2=149 vertices. So I suggest that you check the maximum number of polygons (M) and maximum number of vertices per polygon (N) in your dataset and set the following two parameters accordingly:

--num_queries=M×N \
--num_polys=M \

Hope it helps :)

Martin-1999 commented 9 months ago

Thanks!