当前位置:   article > 正文

yolov9目标检测测试报错AttributeError: ‘list‘ object has no attribute ‘device‘_attributeerror: 'list' object has no attribute 'de

attributeerror: 'list' object has no attribute 'device

报错一:AttributeError: ‘list’ object has no attribute ‘device’

最近微智启软件工作室在运行yolov9目标检测的detect.py测试代码时,报错:
File “G:\down\yolov9-main\yolov9-main\detect.py”, line 102, in run
ValueError: only one element tensors can be converted to Python scalars
pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
File “G:\down\yolov9-main\yolov9-main\utils\general.py”, line 905, in non_max_suppression
device = prediction.device
AttributeError: ‘list’ object has no attribute ‘device’

这是因为general.py代码中,900行左右的代码错误了,可以看到里面是一个包含两个数据的,假如直接设置是会报错的
在这里插入图片描述
general.py的位置可以直接点击报错的这个链接跳转,当然也可以在根目录下的yolov9-main\utils\general.py手动打开
在这里插入图片描述

下面的yolov5的,可以看到只有一个数据,所以不会报错,照抄代码是不对的哟,官方大大!
在这里插入图片描述
所以需要对代码进行遍历,设置它改写后的代码如下,替换之前的代码即可。

  if isinstance(prediction, (list, tuple)):
        processed_predictions = []  # 用于存储处理后的张量列表
        for pred_tensor in prediction:
            # 对每个张量进行处理
            processed_tensor = pred_tensor[0]  # 假设你只关心张量中的第一个结果
            processed_predictions.append(processed_tensor)  # 将处理后的张量添加到列表中

        # 使用处理后的张量列表中的第一个张量作为预测结果
        prediction = processed_predictions[0]

    # 在此之后可以继续使用 prediction 变量
    device = prediction.device
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

替换后的代码格式如下
在这里插入图片描述
之后再运行,即可完美解决!
现在是2024年2月23日,后期官方可能会修改这个bug,根据具体情况来修改。

报错二:AttributeError: ‘list’ object has no attribute ‘view’

Traceback (most recent call last):
  File "G:\down\yolov9-main\yolov9-main\train.py", line 635, in <module>
    main(opt)
  File "G:\down\yolov9-main\yolov9-main\train.py", line 529, in main
    train(opt.hyp, opt, device, callbacks)
  File "G:\down\yolov9-main\yolov9-main\train.py", line 305, in train
    loss, loss_items = compute_loss(pred, targets.to(device))  # loss scaled by batch_size
  File "G:\down\yolov9-main\yolov9-main\utils\loss_tal.py", line 168, in __call__
    pred_distri, pred_scores = torch.cat([xi.view(feats[0].shape[0], self.no, -1) for xi in feats], 2).split(
  File "G:\down\yolov9-main\yolov9-main\utils\loss_tal.py", line 168, in <listcomp>
    pred_distri, pred_scores = torch.cat([xi.view(feats[0].shape[0], self.no, -1) for xi in feats], 2).split(
AttributeError: 'list' object has no attribute 'view'
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

在这里插入图片描述

解决方案:在v8之前,我们都是习惯配置train.py进行训练的,但是根据v9作者在GitHub上的回复来看,我们应该用train_dual.py这个来训练,而不是train.py。

train.py至于有啥用,暂时没见回复,不过我觉得更多的像是一种备份没有删除。

在这里插入图片描述

文章由微智启原创,转载请标明出处,谢谢。
如果还有其他问题,可以联系技术客服:3447362049

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小舞很执着/article/detail/738189
推荐阅读
相关标签
  

闽ICP备14008679号