SarahwXU / HiSup

MIT License
128 stars 18 forks source link

How does offset regression really work? #24

Closed mridvankeles closed 7 months ago

mridvankeles commented 8 months ago

I couldnt really understand the idea and the importance of the junctions and heatmap,offset regressions in this repo.

at the encoder.py ; xint, yint = junctions[:,0].long(), junctions[:,1].long() off_x = junctions[:,0] - xint.float()-0.5 off_y = junctions[:,1] - yint.float()-0.5

this part doing the offset convertion from junction x and y points i see that. And at the inference phase, we use those offsets to get score for nms postprocess. I couldnt find much more details neither on the paper nor on the internet.

ill be preciated if anyone helps. thx

SarahwXU commented 7 months ago

The separate heatmap and offset ensure the accuracy of junction detection for predicting the junction location on the 1/4 resolution feature map. You may refer to papers like ''Holistically-Attracted Wireframe Parsing'' and ''End-to-End Wireframe Parsing".

mridvankeles commented 7 months ago

Now i understand it. Thanks for ur work and reply.