bennetthardwick / darknet.js

A NodeJS wrapper of pjreddie's darknet / yolo.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is possible to limit classes to detect?

Animion opened this issue · comments

To speedup detection is possible to configure classes to detect?

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.