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']