Dears,
Hi, recently, I’m working on object detection. During the post-processing of the object detection, I need to select the valid bounding box which value is higher than threshold.  The code is shown in the following.
c_mask = conf_scores[cl].gt(self.conf_thresh).nonzero().view(-1)
However, nonzero() part is really time consuming in this line. It nearly takes around 17ms to processing 11620 * 20 class result.
Does anyone has any ideas about that?