HRNet OCR 遇到的AttributeError: ‘NoneType‘ object has no attribute ‘shape‘

原创于 2022-01-06 11:36:06 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言 #后端 #机器学习 #计算机视觉

博主在解决代码运行错误时，发现数据集路径设置不正确，特别是缺少了'cityscapes'目录。通过改为绝对路径并修正yaml和cityscapes文件中的路径，问题得到解决。此外，还提到了正在调整预测代码，并计划后续分享。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

总之就是很蠢的问题，作者本身的代码路径有问题，首先数据集路径必须按照官网设置对，注意大小写：
在这里插入图片描述

其次我还是报错了，仔细一看：
在这里插入图片描述

好家伙，路径少了个cityscapes，所以直接用绝对路径，在yaml和cityscapes里改了下代码：

DATASET:
  DATASET: cityscapes
  ROOT: /users/guozibin/5.11.depth-net-train/HRNet-Semantic-Segmentation-HRNet-OCR/HRNet-Semantic-Segmentation-HRNet-OCR/data/cityscapes/
  TEST_SET: '/users/guozibin/5.11.depth-net-train/HRNet-Semantic-Segmentation-HRNet-OCR/HRNet-Semantic-Segmentation-HRNet-OCR/data/list/cityscapes/val.lst'
#都用了绝对路径

 self.root = root
        self.list_path = list_path
        self.num_classes = num_classes

        self.multi_scale = multi_scale
        self.flip = flip
        self.img_list = [line.strip().split() for line in open(list_path)]#这里把root去了
        #self.img_list = [line.strip().split() for line in open(root+list_path)]

        self.files = self.read_files()
        if num_samples:
            self.files = self.files[:num_samples]

然后成功测试：
在这里插入图片描述
另外还有预测的代码，我还在调整，之后会发出来