HPatches源码分解/源码阅读笔记

读取数据

hpatches.py

def get_patch(seq,t,idx):
    """Gets a patch from a sequence with type=t and id=idx"""
    return getattr(seq, t)[idx]

tps = ['ref','e1','e2','e3','e4','e5','h1','h2','h3','h4','h5', 't1','t2','t3','t4','t5']
t in tps
seq=hpatch_sequence(os.path.join(datadir, "hpatches-release", "v_calder"))

class hpatch_sequence:
    """Class for loading an HPatches sequence from a sequence folder"""
    itr = tps
    def __init__(self,base):
        name = base.split(os.path.sep)
        self.name = name[-1]
        self.base = base
        for t in self.itr:
            im_path = os.path.join(base, t+'.png')
            im = cv2.imread(im_path,0)#一条图像
            self.N = im.shape[0]/65
            setattr(self, t, np.split(im, self.N))#切分回一个个图像

计算PR

task.py

def eval_matching(descr,split):
    print('>> Evaluating %s task' % green('matching'))
    start = time.time()

    results = defaultdict(lambda: defaultdict(lambda:defaultdict(dict)))
    pbar = tqdm(split['test'])
    for seq in pbar:
        d_ref = getattr(descr[seq], 'ref')
        gt_l = np.arange(d_ref.shape[0])
        for t in tp:
            for i in range(1,6):
                d = getattr(descr[seq], t+str(i))
                D = dist_matrix(d_ref,d,descr['distance'])
                idx = np.argmin(D,axis=1)#dis最小的系列点idx
                m_l = np.equal(idx,gt_l)#匹配点
                results[seq][t][i]['sr'] = np.count_nonzero(m_l) / float(m_l.shape[0])
                m_d = D[gt_l,idx]#匹配点距离？
                pr,rc,ap = metrics.pr(-m_d,m_l,numpos=m_l.shape[0])
                results[seq][t][i]['ap'] = ap
                results[seq][t][i]['pr'] = pr
                results[seq][t][i]['rc'] = rc
                # print(t,i,ap,results[seq][t][i]['sr'])
    end = time.time()
    print(">> %s task finished in %.0f secs  " % (green('Matching'),end-start))
    return results