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

2021-03-20 02:26

阅读:462

标签:时间   com   win   image   tor   修改   hat   目录   文本   

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

 

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

标签:时间   com   win   image   tor   修改   hat   目录   文本   

原文地址:https://www.cnblogs.com/tonysken/p/12310303.html


评论


亲,登录后才可以留言!