From c52fa4742866125e35e88d0653dac99473e5fffb Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Mon, 16 Apr 2018 10:09:10 +0000 Subject: [PATCH] Loss-graph store automatically (iterations == max_batches) at the end of training --- README.md | 8 +++++--- 1 files changed, 5 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0e81cfd..3eeda63 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ * **OpenCV 3.4.0**: https://sourceforge.net/projects/opencvlibrary/files/opencv-win/3.4.0/opencv-3.4.0-vc14_vc15.exe/download * **or OpenCV 2.4.13**: https://sourceforge.net/projects/opencvlibrary/files/opencv-win/2.4.13/opencv-2.4.13.2-vc14.exe/download - OpenCV allows to show image or video detection in the window and store result to file that specified in command line `-out_filename res.avi` -* **GPU with CC >= 2.0** if you use CUDA, or **GPU CC >= 3.0** if you use cuDNN + CUDA: https://en.wikipedia.org/wiki/CUDA#GPUs_supported +* **GPU with CC >= 3.0**: https://en.wikipedia.org/wiki/CUDA#GPUs_supported ##### Pre-trained models for different cfg-files can be downloaded from (smaller -> faster & lower quality): * `yolov3.cfg` (236 MB COCO **Yolo v3**) - require 4 GB GPU-RAM: https://pjreddie.com/media/files/yolov3.weights @@ -167,7 +167,7 @@ `C:\opencv_3.0\opencv\build\include;..\..\3rdparty\include;%(AdditionalIncludeDirectories);$(CudaToolkitIncludeDir);$(cudnn)\include` - (right click on project) -> Build dependecies -> Build Customizations -> set check on CUDA 9.1 or what version you have - for example as here: http://devblogs.nvidia.com/parallelforall/wp-content/uploads/2015/01/VS2013-R-5.jpg -- add to project all `.c`, `.cpp` & `.cu` files from `\src` +- add to project all `.c` & `.cu` files and file `http_stream.cpp` from `\src` - (right click on project) -> properties -> Linker -> General -> Additional Library Directories, put here: `C:\opencv_3.0\opencv\build\x64\vc14\lib;$(CUDA_PATH)lib\$(PlatformName);$(cudnn)\lib\x64;%(AdditionalLibraryDirectories)` @@ -307,7 +307,9 @@ 9. After training is complete - get result `yolo-obj_final.weights` from path `build\darknet\x64\backup\` - * After each 1000 iterations you can stop and later start training from this point. For example, after 2000 iterations you can stop training, and later just copy `yolo-obj_2000.weights` from `build\darknet\x64\backup\` to `build\darknet\x64\` and start training using: `darknet.exe detector train data/obj.data yolo-obj.cfg yolo-obj_2000.weights` + * After each 100 iterations you can stop and later start training from this point. For example, after 2000 iterations you can stop training, and later just copy `yolo-obj_2000.weights` from `build\darknet\x64\backup\` to `build\darknet\x64\` and start training using: `darknet.exe detector train data/obj.data yolo-obj.cfg yolo-obj_2000.weights` + + (in the original repository https://github.com/pjreddie/darknet the weights-file is saved only once every 10 000 iterations `if(iterations > 1000)`) * Also you can get result earlier than all 45000 iterations. -- Gitblit v1.10.0