解决一些在使用yolov3-tiny过程中出现的bug（Windows）

Posted 2021-03-10 tonysken

这段时间因为搞关于目标检测类型的算法模型，在yolo官网上找到yolov3-tiny模型，这篇博客具体说说调试过程出现的bug

github地址https://github.com/ultralytics/yolov3（这里是框架的官网源码）

首先是关于配置问题

我的是win10+python3.8+pytorch1.4+torchvision0.5+Anaconda

这里要说明一下，torch和torchvision版本不对应会出现此类问题

DLL failed（这是第一个bug）

所以一定要找到相对应的torch和torchvision安装，没有清华镜像的话建议使用Anaconda构建环境，用conda安装，具体安装在pytorch官网上

再来是将框架clone到本地后的代码调试修改

进入根目录下的train.py

如下图

找到这部分代码，可能随着版本的更新会略有差异，如上图划黄线的三行是定义框架，数据集，预训练参数的默认值，更改为自己需要的内容，路径在根路径下

hat.data是我的数据集设置文本，里面包括了训练集和测试集的路径，训练类别，还有一些预定义的内容

cfg是框架

weight是预训练加载好的权重

好了，先写到这，有新的bug再更新