#324 · Primary category: Computer Vision

keras-yolo2

convolutional-networks deep-learning realtime regression yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).

Project last updated:03/24/23

GitHub Stars

1.7K

Forks

770

Contributors

11

License

MIT

Why we included this project

Training a custom object detector usually means wrestling with framework plumbing, but this repo gives you a working YOLOv2 implementation in Keras with a TensorFlow backend, ready to train on your own VOC-format annotations and image folders. You can swap lightweight backbones like MobileNet, SqueezeNet, InceptionV3, or ResNet50, which is handy when you want to trade a little accuracy for faster inference on less powerful hardware. The bundled examples for raccoon, kangaroo, hand, and red blood cell detection double as starting templates, and there is even a raccoon detector demo that runs entirely in the browser. For people who want to see how the training pieces fit together rather than run a black box, the walkthrough of data preparation and training steps is a practical way in.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category