Convenient functions for YOLO v4 based on AlexeyAB Darknet Yolo.

You only look once (YOLO) is a state-of-the-art, real-time object detection system. It is implemented based on the Darknet, an Open Source Neural Networks in C. In this project, I improved the YOLO by adding several convenient functions for detecting objects for research and the development community.

Figure 1. Example of Object Detection using Yolo based on the Darknet.

The added functions are implemented based on AlexeyAB version of Darknet. As it is updated frequently, hereby I publish a stable version of AlexeyAB Darknet Yolo with those convenient functions. This repo will also be updated regularly.

Batch images detector

Figure. The process of batch detecting images in a folder using Yolo based on the Darknet.

The detector function in AlexeyAB Darknet only supports a single image at a time. Therefore I added the batch function into this forked repo, which supports detecting images in a folder in one time. In the meantime, it exports information including the name of the image, the detected classes, the confidence and the bounding box coordinates in JSON and TXT files.

Hope you like it.

Online Demo (on Google Colab)

https://bit.ly/vg_yolo_v4_colab

Github link: https://github.com/vincentgong7/VG_AlexeyAB_darknet

Please also refer to the post for more information:

pjreddie/darknet#723

Update Jan 1, 2021

1. The new version based on AlexeyAB Yolo v4.

2. Compile without change anything on Linux and Windows. Both are tested.

3. Export the bounding box of detected objects in images to JSON.

4. Export the bounding box of detected objects in images to TXT.

5. Added the Google Colab Demo.

Usage

Command

./darknet detector batch cfg/coco.data cfg/yolov4.cfg weights/yolov4.weights io_folder sample_imgs/ output/ -out output/result.json -ext_output > output/result.txt

Parameter explain:

1. The input images are: sample_imgs/

2. The output images are: output/

3. The image name, detected classes, the confidence and the bounding box coordinates is saved in: output/result.txt and in output/result.json

Installation

1. Clone this project, or download this project.

2. Decompress the weight file.

   cd ./weights/

   7z x yolov4.weights.7z.001

   It requires the tool of 7z. You may need to install it if you do not have it.

3. Build the project.

   First of all, go back to the root folder of the project.

   cd ..

   Now you should be in the project root folder of the project, such as: VG_AlexeyAB_darknet

   For linux (e.g. Ubuntu): Make the project with command in the project root folder:

   make

   For windows. It can be successfully built on Visual Studio using project file in the folder of:

   /build/darknet/

4. Use the commandf to batch process images.

./darknet detector batch cfg/coco.data cfg/yolov4.cfg weights/yolov4.weights io_folder sample_imgs/ output/ -out output/result.json -ext_output > output/result.txt

Contact

Any questions please let me know.

If you like it, please also let me know.

vincent.gong7[at]gmail.com

Gong.im

Guestbook