这段时间因为搞关于目标检测类型的算法模型,在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再更新