Skip to content

Constrain similarity matrix with cameraID. #3

@windsor718

Description

@windsor718

In deepreid/identify.py

def match(similarityMatrix, ids, threshold=0.4):
    out = []
    for i in range(similarityMatrix.shape[0]):
        cand_idx = np.where(similarityMatrix[i, :] > threshold)[0]
        print(cand_idx)
        matched = [ids[i]]
        for idx in set(cand_idx) - set([i]):
            print(idx)
            print(similarityMatrix[:, idx])
            max_idx = np.argsort(similarityMatrix[:, idx])[-2]
            if max_idx == i:
                matched.append(ids[idx])
        out.append(matched)
    return out

May need to be constrained by camera id to improve accuracy.
May not be necessary after model re-training?

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions