search

mindspore-lab / mindyolo

A toolbox of yolo models and algorithms based on MindSpore
Apache License 2.0
100 stars 42 forks source link

在310B上MindX推理报错AttributeError: 'COCODataset' object has no attribute 'column_names_getitem' #239

Closed langshhhh closed 11 months ago

langshhhh commented 11 months ago

测试环境： cann6.2.rc2 python3.9.2 mindxsdk 5.0.rc2 测试方式： 从官方网站下在了yolov8m的模型，直接运行如下命令， python ./deploy/test.py --model_type MindX --model_path /root/yolo.om --config ./configs/yolov8/yolov8n.yaml 报错AttributeError: 'COCODataset' object has no attribute 'column_names_getitem' 数据集使用的是coco数据集

zhanghuiyao commented 11 months ago

看mindyolo r0.2/master 代码是有这个属性的，看看环境中是否有安装了mindyolo 0.1或有PATH指向了其他包

langshhhh commented 11 months ago

装了mindyolo 0.2 只配置了cann的变量 没有其他PATH了

zhanghuiyao commented 11 months ago

看起来还是没找到这个属性，可以在代码运行的地方打断点调试下，看看是不是用到了当前目录下的COCODataset

langshhhh commented 11 months ago

之前 翻了代码 是有这个属性 但就是找不到不知道为什么