Skip to content

linxi1158/yolo-windows

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Yolo-Windows v1

"You Only Look Once: Unified, Real-Time Object Detection"

Here is a Yolo v1 windows version (for object detection)

Contributtors: https://github.com/pjreddie/darknet/graphs/contributors

This repository is forked from: https://github.com/frischzenger/yolo-windows Which is forked from Linux-version: https://github.com/pjreddie/darknet

More details: http://pjreddie.com/darknet/yolov1/

Requires:
Pre-trained models for different cfg-files can be downloaded from (smaller -> faster & lower quality):
Examples of results:

Everything Is AWESOME

Others: https://www.youtube.com/channel/UC7ev3hNVkx4DzZ3LO19oebg

Example of usage in cmd-files:
  • darknet.cmd - initialization with 1 GB model yolo.weights & yolo.cfg and get command prompt: enter the jpg-image-filename
  • darknet_demo.cmd - initialization with 1 GB model yolo.weights & yolo.cfg and play your video file which you must rename to: test.mp4
  • darknet_demo_small.cmd - initialization with 359 MB small model yolo-small.weights & yolo-small.cfg and play your video file which you must rename to: test.mp4
  • darknet_net_cam.cmd - initialization with 1 GB model, play video from network video-camera mjpeg-stream and store result to: test_dnn_out.avi
For using network video-camera mjpeg-stream with any Android smartphone:
  1. Download for Android phone mjpeg-stream soft: IP Webcam / Smart WebCam

Smart WebCam - preferably: https://play.google.com/store/apps/details?id=com.acontech.android.SmartWebCam IP Webcam: https://play.google.com/store/apps/details?id=com.pas.webcam

  1. Connect your Android phone to computer by WiFi (through a WiFi-router) or USB
  2. Start Smart WebCam on your phone
  3. Replace the address below, on shown in the phone application (Smart WebCam) and launch:
darknet.exe yolo demo yolo.cfg yolo.weights http://192.168.0.80:8080/video?dummy=param.mjpg -i 0

Windows (Visual Studio) Support Now yolo supports windows with Visual Studio. Just change the include directories / library directories to your own ones, and compile the codes with X64 Release mode. You may need pthread-windows while compiling and linking

How to train yourself:

To training for your 20 objects download pre-trained convolutional layers:

Then use the following command lines:

  • For tiny-model: darknet yolo train yolo-tiny.cfg darknet.conv.weights
  • For small-model: darknet yolo train yolo-small.cfg extraction.conv.weights
  • For full-model: darknet yolo train yolo.cfg extraction.conv.weights

To train for a different number of objects (not 20) or for different number of sides (not 7), you should change your .cfg-file 3 lines:

  • output= (5 x 2 + C) x S x S
  • classes= C
  • side= S

Where:

  • C = number of objects (by default 20)
  • S = number of sides (by default 7)

More details: http://pjreddie.com/darknet/yolov1/

How to compile:

  1. If you have CUDA 7.5, OpenCV 2.4.9 (C:\opencv_2.4.9) and MSVS 2013 (or 2015) then start MSVS, open yolo-windows\build\darknet\darknet.sln and do the: Build -> Build darknet

  2. If you have other version of CUDA (not 7.5) then open yolo-windows\build\darknet\darknet\darknet.vcxproj by using Notepad, find 2 places with "CUDA 7.5" and change it to your CUDA-version, then do step 1

  3. If you have other version of OpenCV 2.4.x (not 2.4.9) then you should change pathes after \darknet.sln is opened

3.1 (right click on project) -> properties -> C/C++ -> General -> Additional Include Directories

3.2 (right click on project) -> properties -> Linker -> General -> Additional Library Directories

  1. If you have other version of OpenCV 3.x (not 2.4.x) then you should change many places in code by yourself.

How to compile (custom):

Also, you can to create your own darknet.sln & darknet.vcxproj, this example for CUDA 7.5 and OpenCV 2.4.9

Then add to your created project:

  • (right click on project) -> properties -> C/C++ -> General -> Additional Include Directories, put here:

C:\opencv_2.4.9\opencv\build\include;..\..\..\3rdparty\include;%(AdditionalIncludeDirectories);$(CudaToolkitIncludeDir)

C:\opencv_2.4.9\opencv\build\x64\vc12\lib;$(CUDA_PATH)lib\$(PlatformName);%(AdditionalLibraryDirectories)

  • (right click on project) -> properties -> Linker -> Input -> Additional dependecies, put here:

..\..\..\3rdparty\lib\x64\pthreadVC2.lib;cublas.lib;curand.lib;cudart.lib;%(AdditionalDependencies)

  • (right click on project) -> properties -> C/C++ -> Preprocessor -> Preprocessor Definitions

OPENCV;_CRT_SECURE_NO_WARNINGS;GPU;WIN32;NDEBUG;_CONSOLE;_LIB;%(PreprocessorDefinitions)

  • compile to .exe (X64 & Release) and put .dll`s near with .exe:

pthreadVC2.dll, pthreadGC2.dll from yolo-windows\3rdparty\dll\x64

cusolver64_75.dll, curand64_75.dll, cudart64_75.dll, cublas64_75.dll - 75 for CUDA 7.5 or your version, from C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5\bin

About

a yolo windows version(for object detection)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C 90.9%
  • Cuda 8.2%
  • Other 0.9%