search

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

New utility feature class filtering for Object Detectors 2D #467

Closed katdimitris closed 9 months ago

katdimitris commented 10 months ago

Implemented class filtering for object detectors using a wrapper. The wrapper accepts a Learner object and a list of desired classes for detection. These classes should be within the set that the learner is capable of detecting. An inference demo for YOLOv5 is also available.

Closes #464.