Skip to content

shekkizh/VISSL_NNK_Benchmark

Repository files navigation

VISSL-NNK: Self Supervised Model Evaluation

Blog Post

NNK evaluation of Self supervised models. The source code is integrated with VISSL for feature extraction and can be applied to any dataset and model that can be loaded by VISSL. One can also, perform evaluation by directly using the relevant code corresponding to NNK classifier on pre-extracted features without VISSL.

Requirements

The source code assumes the following packages are installed:

  • VISSL: This forms the bacbone for retrieving features of various self supervised models. The code was tested with installation from VISSL source.
  • FAISS: Used to query for nearest neighbors. Code was tested with pip installed gpu version of package. Installation instructions for both packages can be found here and here

Data and Pretrained models

The config files provided assume evaluation on ImageNet data which need to be downloaded and saved to a local directory. Further instructions on how to setup the dataset can be found in VISSL documentation.

VISSL documentation also provides instruction and tutorials on loading pretrained self supervised models for feature extraction. The config files in this repository provide examples for some of the recently popular self supervised models.