windows11跑通YOLOX踩坑总结

我的配置

:显卡NVIDIA 4080;CUDA12.1;
##################################

我的提前下载


YOLOX源码https://github.com/Megvii-BaseDetection/YOLOX
权重放在源码文件里自己新建weights文件中,下载链接如下:
1.yolox_nano:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_nano.pth
2.yolox_tiny:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_tiny.pth
3.yolox_s:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_s.pth
4.yolox_m:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_m.pth
5.yolox_l:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_l.pth
6.yolox_x:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_x.pth
7.yolox_darknet53:https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_darknet.pth
conda的镜像源问题:教程参考
########################################
**


2024.4.11

我的步骤

**:
anaconda创建环境→→→→→→→→→→→→→→→
进入环境conda activate xxxx→→→→→→→→→→→
pytorch官网https://pytorch.org/get-started/locally/根据自己配置安装pytorch→→→→→→→→→→→→→→→→→→
安装完成后核查:
>>pip list #查看环境内的包。
>>python #查看是否能正常使用。
>>import torch #查看是否能正常调用pytorch包。
>>torch.cuda.is_available() #如果返回True,则表示能够正常使用cuda。FALSE说明你下载的是CPU版本
>>exit() #退出模式。→→→→→→→→→→→→→→→→→→→→→→→→
pycharm打开YOLO源码项目→→→→→→→→→→→
根据YOLOX\requirements.txt文件要求安装工具包:
安装方法多样
直接根据requirements.txt文件下载法:

pip install -U pip && pip3 install -r requirements.txt

我是自己在pycharm的python解析器一个一个手动下载,下载期间也有些问题,这个后续填坑
→→→→→→→→→→→→→→→→→→→→→→→→
满足所有软件包后,首先cd到你的源文件夹,cd无效就pushd,我都是pushd才ok

cd YOLOX  #pushd YOLOX
python setup.py develop # 配置你的YOLOX模型,要先放权重进去

→→→→→→→→→→→→→→→→→→→→→→→→
测试demo:都是找的大家给的测试命令,具体为什么跑不通我还没仔细研究,留坑

#测试命令行得通:
python tools/demo.py image -n yolox-nano -c weights/yolox_nano.pth --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device gpu

#我的行不通:
python tools/demo.py image -f exps/default/yolox_s.py -c ./yolox_s.pth --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device gpu
python tools/demo.py image -n yolox-s -c weights/yolox_s.pth.tar --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result
python3 tools/demo.py image -n yolox-s -c yolox_s.pth.tar --path assets/dog.jpg --conf 0.3 --nms 0.5 --tsize 640 --save_result --device [gpu]

出图在YOLOX\YOLOX_outputs
在这里插入图片描述
→→→→→→→→→→→→→→→→→→→→→→→→
换数据集VOC:又都是坑坑,直接上链接
→→→→→→→→→→→→→→→→→→→→→→→→
数据集操作我跟的视频教程YOLO-X(yolox)训练自己的数据集
→→→→→→→→→→→→→→→→→→→→→→→→
YoloX训练报错“IndexError: list index out of range
原因:简单来说就是代码基于linux使用“/”

而在Python了里windows系统路径表示是"\\"在这里插入图片描述
→→→→→→→→→→→→→→→→→→→→→→→→
YOLOX报错FileNotFoundError: [Errno 2] No such file or directory: ‘xxx.xml‘
→→→→→→→→→→→→→→→→→→→→→→→→
留几个疑惑,路过的大佬欢迎帮忙解答一下哈:
我不太理解代码下载在文件里不在conda环境里,为啥要用到conda建的环境下的这个voc.py文件啊!!!?我之前一直以为是路径设置问题,没想到这个文件确实需要
在这里插入图片描述
还有这个攻略,修改函数这里,我是只更改了数据集路径(因为使用自带函数他会找你的所有盘然后找到的第一个同名文件就直接赋值使用,很容易错),还删除了2012的数据代码
在这里插入图片描述
在这里插入图片描述
针对这里还有些疑惑,我跟的视频教程YOLO-X(yolox)训练自己的数据集,他的YOLOX\exps\example\yolox_voc\yolox_voc_s.py是这样的,up主也可以跑通的,但是python小白怕改错了就按照上面的自己的改的,没换函数,也跑通啦~
在这里插入图片描述
在这里插入图片描述
就跑了个训练哈哈哈,跑通我就是快乐小狗嘿嘿在这里插入图片描述


2024.4.12
用训练出的最后模型跑一张图测试

python tools/demo.py image -f exps/example/yolox_voc/yolox_voc_nano.py -c YOLOX_outputs/yolox_voc_nano/best_ckpt.pth --path path/to/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --device gpu

在这里插入图片描述
!!!重新划分数据集后,先删除datasets/VOCdevkit下的annotations_cache、results文件夹后再训练

  • 28
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值