Skip to content

Get started

Create Dataset

Create toy dataset

python tools/dataset/extract_torchvision_dataset.py --dataset CIFAR100 --toy ../datasets/cifar100

For custom datasets, set the format as follows:

root/
    train/
        cate_1/
            xxx.jpg
        cate_2/
            xxx.jpg
    val/
        cate_1/
            xxx.jpg
        cate_2/
            xxx.jpg

Extract Features

python extract.py --arch resnet18 --data toy.yaml

After completing the extraction, you can view the file in runs/extract/expXXX/

Retrieval Features

python retrieval.py /path/to/gallery.pkl /path/to/query.pkl

After completing the retrieval, the retrieval results will be saved in runs/retrieval/expXXX/info.pkl

Metric Features

python metric.py /path/to/retrieval.pkl --evaluate ACCURACY

Demo: toy dataset

# Create toy dataset
python tools/dataset/extract_torchvision_dataset.py --dataset CIFAR100 --toy ../datasets/cifar100
# Extract
python extract.py --arch resnet18 --data toy.yaml
# Retrieval
python retrieval.py runs/extract/exp/gallery.pkl runs/extract/exp/query.pkl
# Metric
python metric.py runs/retrieval/exp/info.pkl --evaluate ACCURACY
...
Accuracy: ['19.400', '34.300', '45.100', '58.400']