Skip to content

Light version of convolutional neural network Yolo v3 & v2 for objects detection with a minimum of dependencies (INT8-inference, BIT1-XNOR-inference)

License

Notifications You must be signed in to change notification settings

AlexeyAB/yolo2_light

Repository files navigation

yolo2_light

Light version of convolutional neural network Yolo v3 & v2 for objects detection with a minimum of dependencies (INT8-inference, BIT1-XNOR-inference)

This repository supports:

  • both Windows and Linux
  • both OpenCV <= 3.3.0 and OpenCV 2.4.13
  • both cuDNN >= 7.1.1
  • CUDA >= 8.0

How to compile:

  • To compile for CPU just do make on Linux or build yolo_cpu.sln on Windows

  • To compile for GPU set flag GPU=1 in the Makefile on Linux or build yolo_gpu.sln on Windows

    Required both CUDA >= 8.0 and cuDNN >= 7.1.1

How to start:

  • Download yolov3.weights to the bin directory and run ./yolo.sh on Linux (or yolo_cpu.cmd / yolo_gpu.cmd on Windows)
  • Download yolov3-tiny.weights to the bin directory and run ./tiny-yolo.sh

How to use INT8-inference:

How to use BIT1-XNOR-inference - only for custom models (you should train it by yourself):

  • You should base your cfg-file on yolov3-spp_xnor_obj.cfg and train it by using this repository as usual https://github.com/AlexeyAB/darknet by using pre-trained file darknet53_448_xnor.conv.74
  • Then use it for Detection-test or for getting Accuracy (mAP):
    • ./darknet detector test data/obj.names yolov3-spp_xnor_obj.cfg data/yolov3-spp_xnor_obj_5000.weights -thresh 0.15 dog.jpg
    • ./darknet detector map data/obj.data yolov3-spp_xnor_obj.cfg data/yolov3-spp_xnor_obj_5000.weights -thresh 0.15

Other models by the link: https://pjreddie.com/darknet/yolo/

About

Light version of convolutional neural network Yolo v3 & v2 for objects detection with a minimum of dependencies (INT8-inference, BIT1-XNOR-inference)

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published