D-FINE icon indicating copy to clipboard operation
D-FINE copied to clipboard

Filter out desired classes

Open AnishMathewOommen opened this issue 11 months ago • 0 comments

Hi there, thank you for your great work!

I was wondering if there is any way when using a pretrained model to filter out detection results based on the class ID e.g. if I only wanted the detector to output human detections I pass it class ID of 1 ?

I have implemented this post detection, but I am just checking if there is some already existing functionality.

AnishMathewOommen avatar Jan 21 '25 10:01 AnishMathewOommen