总之就是很蠢的问题,作者本身的代码路径有问题,首先数据集路径必须按照官网设置对,注意大小写:
其次我还是报错了,仔细一看:
好家伙,路径少了个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]
然后成功测试:
另外还有预测的代码,我还在调整,之后会发出来