- dataLoad文件夹:加载数据集
- datasets文件夹:存放你自己的数据集,包括训练集和测试集,自己分好。二级目录里面存放的就是jpg文件和xml文件。
- model_data文件夹: 训练需要用到的文件。其中需要修改的就是类别文件your_classes
- net文件夹:网络结构
- train.py:训练文件
- 还没有写预测文件,先记录一下
- 准备数据集,标注,并放入 二 、解释 中所讲的文件夹下
- 修改model_data中的your_classes.txt为你自己的类别,一行一个
- 在train.py下的参数设置中修改你自己的路径,其实也就是类别数量需要改一下,如果你不改类别的文件名的话
- 如果你是cpu的话,注释掉其中一句。如果是单机多卡的话,注释里面也写了怎么训练。如果是只有一块卡的话,直接运行训练即可。
paddle-yolov3: 这是一个用paddle实现yolov3的项目,可用来训练自己的数据 (gitee.com)https://gitee.com/hablee/paddle-yolov3



