Using Yolo and Darknet as the underlying codebase
Used :-Darknet,Darknet Architecture,OpenCV
Refrence:- Custom object training and detection with YOLOv3, Darknet and OpenCV
NOTE !! : Before running you should have a DIRECORY called “training” (without he quotes) that contains all the images with the respictive txt file (having the same name ,i.e for an image image1.png the text file will be named image1.txt) and a classes.txt file (contains all the classifier names). The txt files can be generated through various softwares, but for our work we had used LabelImg.
Run config-1.sh only once
It will download the necessary files from the darknet git hub repository and also a pre-trained architecture from darknet’s official website darknet53.conv.74 .(Total Download 166 MB.)
Run config-2.sh
It should be run whenever an update is made to the training directory.It collects all the new information and updates the necessay files that will be created inside darknet to train or test.
Run config-3.sh
To start the training process. (Make sure to run this again if any updates to the data sets are made and new weights are required.
By this time you should have got a waight file in the backup folder. now to go further you need to have sample folder. inside that you must have only one image that will be used for testing as test data.
Run config-4.sh
To try detect an image from the trained data.
inside config-4 we initailly need to insert
- the image path which is going to be tested
- the yolo custom cfg file path
- the weights path which we created
- the custom names file consist of all classes
./config-1.sh
./config-2.sh
./config-3.sh
./config-4.sh
config-1 , config-2 , config-3 , config-4
clone the darknet repository https://github.com/pjreddie/darknet
then get default weight files from https://pjreddie.com/media/files/darknet53.conv.74