pycharm训练数据慢_yolov4体验(keras模型训练+检测)亲测可行

1.环境准备
2.代码准备
3.2007_voc数据准备&预处理
4.模型训练
5.检测展示https://blog.csdn.net/ly_twt/article/details/105761312https://github.com/Ma-Dan/keras-yolo4https://arxiv.org/pdf/2004.10934.pdf
----------------------------------------------------------------------------------------------------------------------
1.环境配置cpu版（python3.6）

2.代码准备https://github.com/Ma-Dan/keras-yolo4
git clone https://github.com/Ma-Dan/keras-yolo4.git

*注意 yolo4.weights 单独下载 https://drive.google.com/uc?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT&export=download
下载完成后 执行 python convert.py进行转换 自动生成 yolo4_weights.h5

3.2007_voc数据准备&预处理

wget https://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar

wget https://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar

tar xf VOCtrainval_06-Nov-2007.tar

tar xf VOCtest_06-Nov-2007.tar
按照格式放置文件

#修改1 训练自己的数据类别

#修改2 训练自己的数据类别

然后执行 python voc_annotation.py 会在 下面的文件夹路径下生成3个txt 文件

防止内存不够 修改 batch_size

4.模型训练
执行 python train.py

CPU 训练慢，仅效果展示

5.检测展示
将测试文件放到 下面路径
执行 python test.py

原图

展示效果