yolov3训练自己的模型和进行预测

昨天搞定了数据集，今天来训练模型。

Pytorch 搭建自己的YOLO3目标检测平台（Bubbliiiing 深度学习 教程）_哔哩哔哩_bilibili

第一步：制作训练集。

第二步：运行voc_annotation.py

这步的目的是生成一些训练所需的txt文件。

ImageSets文件夹下的是划分训练集、验证集合测试集的比例。还有根目录下的2007train.txt和2007val.txt

第三步：调整train.py参数。 

我没cuda，改了false。

然后类别bdd100k:

 其它的对着源码调就行，注释很详细。

Python-ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed.

报错1：

Python-ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed.

解决方法：

pip install tensorboard

报错2：

Import Error: no module named 'past'

解决方法：

pip3 install future

报错3：

Caught FileNotFoundError in DataLoader worker process 0.

解决方法：

我在别人的csdn那看到的说法是：标签路径不能带空格。

如图：

后来我把我的文件夹名字改成了graduation-project，然后又跑了一下voc_annotation.py和train.py就可以了。

由于我电脑是小天才学习电脑，后面预测部分暂时先不做了，之后借别人电脑试试。

报错4：

具体错误忘截图了，大概就是用自己数据集跑voc_annotation.py说找不到trainval.txt

解决方法：

仿造那个voc数据集下的文件夹形式，在ImageSets文件夹下创建Main文件夹，再创建四个txt

使用自己的模型进行预测

简单来说，就是就修改yolo.py的参数（特别是路径），然后去predict.py中修改模式即可。