yolox 训练错误记录

1.(未解决)AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'

 并且是anaconda prompt shell里面运行没错,pycharm训练就出这个错!

2.(已经解决)

由于yolox还使用我之前用的数据集缓存与我更换后的数据集对应不上导致的。由此找到了缓存文件annots.pkl文件的所在地址:datasets/VOCdevkit/annotations_cache直接将该文件夹删除(yolox再次运行时如果找不到该文件夹会再次新建重新缓存数据),再次运行训练命令开始训练后该问题没有再次出现。


3.2022-10-26 23:06:35 | ERROR    | yolox.core.launch:98 - An error has been caught in function 'launch', process 'MainProcess' (5328), thread 'MainThread' (382

最后报错FileNotFoundError: [Errno 2] No such file or directory: 'G101.xml'

 

 解决:将yolox/data/datasets/voc.py文件修改为如下图所示:

 

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
Yolox (You Only Look Once) 是一种基于目标检测的深度学习模型,通常在计算机视觉领域被广泛应用。VisDrone是一个开放式的航拍无人机数据集,其中包含了丰富的目标类别和复杂的场景。那么如何用Yolox来训练VisDrone呢? 首先,我们需要将VisDrone数据集进行预处理,确保图像尺寸的一致性以及标注信息的正确性。可以使用Python的图像处理库如OpenCV来完成这一步骤。 接着,我们需要使用Yolox模型进行训练Yolox模型基于PyTorch深度学习框架,因此我们需要安装相应的库并配置环境。然后,我们可以将VisDrone数据集分为训练集和验证集,并用Yolox提供的训练脚本进行训练。该脚本可以自动完成模型的训练迭代,以及模型在验证集上的性能评估。 在训练过程中,我们可以根据需要调整Yolox模型的超参数,例如学习率、迭代次数等,以提高模型的性能。此外,Yolox还支持多种数据增强技术,如随机裁剪、图像旋转等,可以进一步提升模型的鲁棒性和泛化能力。 训练完成后,我们可以使用Yolox模型对新的VisDrone图像进行目标检测。通过加载训练好的模型权重,并利用Yolox提供的推理脚本,我们可以快速准确地检测出图像中的目标,并获取其位置和类别信息。 综上所述,使用Yolox模型训练VisDrone数据集可以帮助我们实现高效准确的目标检测。通过合理调整超参数和应用数据增强技术,我们可以进一步提升模型的性能。这种方法不仅适用于VisDrone数据集,也可以应用于其他类似的航拍无人机数据集,为各种实际应用场景提供强大的目标检测能力。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值