km1414 / CNN-models

YOLO-v2, ResNet-32, GoogLeNet-lite

CNN-models

YOLO-v2

YOLO v2 model for objects detection from image, video and live stream from webcam. Runs on pre-trained weights, GPU is preferred, CPU works as well.

Papers:

• https://arxiv.org/pdf/1506.02640.pdf
• https://arxiv.org/pdf/1612.08242.pdf

Website:

• https://pjreddie.com/darknet/yolo/
  

Setup instructions

ResNet-32

ResNet-32 model for CIFAR-10 image recognition.
Keras. Python 3.
Paper: https://arxiv.org/pdf/1512.03385.pdf
Some results:

GoogLeNet-lite

GoogLeNet-lite (Inception_v1) model for CIFAR-10 image recognition.
Keras. Python 3.
Paper: https://arxiv.org/pdf/1409.4842.pdf
Some results: