CLRNet
CLRNet copied to clipboard
about nms for several categories
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 Did you find a solution? I have similar issue. Thanks!
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!