Use Your Data To Train
The original instruction of retraining YOLOv2 model see github repository of darknet
In the current of BerryNet, TinyYolo is used instead of YOLOv2. The major differences are:
- Create file yolo-obj.cfg with the same content as in
- Download pre-trained weights of darknet reference model,
darknet.weights.12, for the convolutional layers (6.1MB)
The rest parts are the same as retraining YOLO.
If you use LabelMe to annotate data,
utils/xmlTotxt.py can help convert the xml format to the text format that darknet uses.