search

bennetthardwick / darknet.js

A NodeJS wrapper of pjreddie's darknet / yolo.
65 stars 27 forks source link

Is possible to limit classes to detect? #13

Closed Animion closed 6 years ago

Animion commented 6 years ago

To speedup detection is possible to configure classes to detect?

bennetthardwick commented 6 years ago

Unfortunately no. The only way that you could limit the amount of classes to detect and get a speed boost, would be to train a new model that only has those classes. If you're interested in doing this, have a look here. This library uses pjreddie's darknet under the hood, so it's limited only by what it can do.