-
2019-1-19发布的YOLO-V3-IOU项目,基于Keras和TensorFlow,专注于动漫人脸检测的YOLO3算法实现
资源介绍
YOLO-V3-IOU
YOLO3 动漫人脸检测 2019-1-19
环境
tensorflow-gpu 1.12.0
Keras 2.2.4
数据集的准备
数据集标签制作工具下载:
运行prepare.py将数据集划为训练集,验证集和测试集
训练模型
加载权重,将权重h5文件放入models文件夹
最新权重下载地址:
若重新开始训练,将load_pretrained置为False
设置学习率,若重新训练,建议设0.001,若加载权重,建议设较小的学习率
运行train.py
预测
加载权重,将训练好的权重h5文件放入models文件夹
运行run.py,对数据集进行预测输出,输出在outputs文件夹中
预测新的图片
from predict import YOLO
from PIL imp
- 上一篇: THE PH.D. GRIND.pdf
- 下一篇:没有了