0
点赞
收藏
分享

微信扫一扫

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

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

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

好家伙,路径少了个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]

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

举报

相关推荐

0 条评论