search

hollance / Forge

A neural network toolkit for Metal
MIT License
1.27k stars 172 forks source link

How do I only use 1 or 2 classes for the YOLO example? #29

Closed EriFisher closed 6 years ago

EriFisher commented 6 years ago

Hi I love the framework. How do I make it so yolo won't use every class, for instance so it will work on just humans?

hollance commented 6 years ago

You can simply ignore the bounding boxes for the non-human classes, or retrain YOLO so that it will only recognize humans and nothing else.