def __init__(self, root):
super(pcd_eval, self).__init__()
self.img_t0_root = pjoin(root, 't0')
self.img_t1_root = pjoin(root, 't1')
self.img_mask_root = pjoin(root, 'mask')
self.filename = list(spt(f)[0] for f in os.listdir(self.img_mask_root) if check_validness(f))
self.filename.sort()
def check_validness(f):
return any([i in spt(f)[1] for i in ['jpg','png']])
def __init__(self, root):
super(vl_cmu_cd_eval, self).__init__()
self.img_root = pjoin(root, 'RGB')
self.img_mask_root = pjoin(root, 'GT')
self.filename = list(spt(f)[0] for f in os.listdir(self.img_mask_root) if check_validness(f))
self.filename.sort()