Object detectors 2D class filtering

opendr-eu / opendr

A modular, open and non-proprietary toolkit for core robotic functionalities by harnessing deep learning

Apache License 2.0

611 stars 94 forks source link

Object detectors 2D class filtering #464

Closed tsampazk closed 9 months ago

tsampazk commented 10 months ago

Create a learner wrapper class to filter the detection output of an object detector's infer method. This class should be initialized providing a valid subset of specific classes that the user wants to detect.