YOLO3 动漫人脸检测 2019-1-19
- tensorflow-gpu 1.12.0
- Keras 2.2.4
- 如果运行时出现 Blas SGEMM launch failed 错误，请参考qqwweee/keras-yolo3#362
- 数据集标签制作工具下载：https://github.com/tzutalin/labelImg
- 运行prepare.py将数据集划为训练集，验证集和测试集
- 加载权重，将权重h5文件放入models文件夹
- 最新权重下载地址：https://pan.baidu.com/s/1QKODAhRU_Rw5lpf8NcIR4g
- 若重新开始训练，将load_pretrained置为False
- 设置学习率，若重新训练，建议设0.001，若加载权重，建议设较小的学习率
- 运行train.py
- 加载权重，将训练好的权重h5文件放入models文件夹
- 运行run.py，对数据集进行预测输出，输出在outputs文件夹中
- 预测新的图片
- from predict import YOLO
- from PIL import Image
- yolo = YOLO()
- img = Image.open(r'./test.jpg')
- output = yolo.detect_image(img)
- output.show()
