CLRNet icon indicating copy to clipboard operation
CLRNet copied to clipboard

about nms for several categories

Open GeorgeFanFan opened this issue 2 years ago • 2 comments

Dear author, if i changed the predictions[:3] for 3 categories classisfication which the third label is background, how can i change the nms code in clr_head/get_lanes? Do i need to change the nms_kernel.cu such as prop size?

GeorgeFanFan avatar Jun 23 '22 03:06 GeorgeFanFan

@GeorgeFanFan Did you find a solution? I have similar issue. Thanks!

sahamitul avatar Jul 20 '22 21:07 sahamitul

Dear author, if i changed the predictions[:3] for 3 categories classisfication which the third label is background, how can i change the nms code in clr_head/get_lanes? Do i need to change the nms_kernel.cu such as prop size?

Did you find a solution? I have similar issue. Thanks!

UcasCore avatar Apr 26 '23 07:04 UcasCore