line 97 in hawp/parsing/detector.py

cherubicXN / hawp

Holistically-Attracted Wireframe Parsing [TPAMI'23] & [CVPR' 20]

MIT License

296 stars 52 forks source link

line 97 in hawp/parsing/detector.py #28

Closed chenmingxiang110 closed 2 years ago

chenmingxiang110 commented 2 years ago

line 97 in hawp/parsing/detector.py where ".reshape(128,-1,32)" should probably change to ".reshape(self.dim_loi, -1, self.n_pts0)" in case the dim_loi and n_pts0 are changed to different values.