- use YOLOv3 training with own data
- Compile YOLOv3 training on GPU
- A Keras implementation of YOLOv3 detection inspired by qqwweee/keras-yolo3.
see YOLOv3训练自己的数据 for details.
see darknet YOLO 编译使用GPU for details.
training with own data(rock climbers,classes of male and female)
- Download trained YOLOv3 weights from baidu yun.
- Convert the Darknet YOLO model to a Keras model.
- Run YOLO detection.
python convert.py ./cfg/yolov3.cfg yolov3.weights yolo.h5
python yolo.py
PS:
- change value of 'self.score' in yolo.py to adjust detection threshold
- In my case, I trained 400 epoch with 200 pictures,it is far from enough to obtain a good test result.
- some examples: