tracel-ai/burn

Boolean mask indexing

Opened this issue · 0 comments

Feature description

Boolean mask indexing, similar to:

inds = scores > self.conf_threshold
boxes, landmarks, scores = boxes[inds], landmarks[inds], scores[inds]

From discord