zhanggang001 / HEDNet

HEDNet (NeurIPS 2023) & SAFDNet (CVPR 2024 Oral)
Apache License 2.0
88 stars 7 forks source link

About the detection head #14

Closed lebron-2016 closed 1 week ago

lebron-2016 commented 1 month ago

Hi, when predicting the final heatmap and object size, it looks like you are using normal convolution. Why not use sparse convolution here?

image

zhanggang001 commented 1 month ago

Hi, when predicting the final heatmap and object size, it looks like you are using normal convolution. Why not use sparse convolution here?

image

It uses Conv1d, applied to the queries, not to dense feature maps.