S623333的博客

原创 yolo目标检测项目-人脸识别

代码会训练数据集300次，每10次会生成1个模型，还会自动保存效果最优模型best_epoch_weights.pth（存储路径为yolov4-pytorch-master\logs\best_epoch_weights.pth），此外还生成了epoch_loss.png损失曲线图和epoch_map.png平均精度曲线图。之后运行voc_annotation.py文件生成2007_train.txt和2007_val.txt。模型预测需要用到yolo.py文件和predict.py文件。