mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
18 lines
539 B
Python
18 lines
539 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import torch
|
|
|
|
from mmselfsup.evaluation import knn_classifier
|
|
|
|
|
|
def test_knn_classifier():
|
|
train_feats = torch.ones(200, 3)
|
|
train_labels = torch.ones(200).long()
|
|
test_feats = torch.ones(200, 3)
|
|
test_labels = torch.ones(200).long()
|
|
num_knn = [10, 20, 100, 200]
|
|
for k in num_knn:
|
|
top1, top5 = knn_classifier(train_feats, train_labels, test_feats,
|
|
test_labels, k, 0.07)
|
|
assert top1 == 100.
|
|
assert top5 == 100.
|