Home

Awesome

LearningToCompare_ZSL

PyTorch code for CVPR 2018 paper: Learning to Compare: Relation Network for Few-Shot Learning (Zero-Shot Learning part)

For Few-Shot Learning part, please visit here.

Requirements

Python 2.7

Pytorch 0.3

Data

Download data from here and unzip it unzip data.zip.

Run

ZSL and GZSL performance evaluated under GBU setting [1]: ResNet feature, GBU split, averaged per class accuracy.

AwA1_RN.py will give you ZSL and GZSL performance on AwA1 with attribute under GBU setting [1].

AwA2_RN.py will give you ZSL and GZSL performance on AwA2 with attribute under GBU setting [1].

CUB_RN.py will give you ZSL and GZSL performance on CUB with attribute under GBU setting [1].

ModelAwA1 T1usHCUB T1usH
DAP [2]44.10.088.70.040.01.767.93.3
CONSE [3]45.60.488.60.834.31.672.23.1
SSE [4]60.17.080.512.943.98.546.914.4
DEVISE [5]54.213.468.722.452.023.853.032.8
SJE [6]65.611.374.619.653.923.559.233.6
LATEM [7]55.17.371.713.349.315.257.324.0
ESZSL [8]58.26.675.612.153.912.663.821.0
ALE [9]59.916.876.127.554.923.762.834.4
SYNC [10]54.08.987.316.255.611.570.919.8
SAE [11]53.01.877.13.533.37.854.013.6
DEM [12]68.432.884.747.351.719.657.929.2
RN (OURS)68.231.491.346.755.638.161.447.0
ModelAwA2 T1usH
DAP [2]46.10.084.70.0
CONSE [3]44.50.590.61.0
SSE [4]61.08.182.514.8
DEVISE [5]59.717.174.727.8
SJE [6]61.98.073.914.4
LATEM [7]55.811.577.320.0
ESZSL [8]58.65.977.811.0
ALE [9]62.514.081.823.9
SYNC [10]46.610.090.518.0
SAE [11]54.11.182.22.2
DEM [12]67.130.586.445.1
RN (OURS)64.230.093.445.3

Citing

If you use this code in your research, please use the following BibTeX entry.

@inproceedings{sung2018learning,
  title={Learning to Compare: Relation Network for Few-Shot Learning},
  author={Sung, Flood and Yang, Yongxin and Zhang, Li and Xiang, Tao and Torr, Philip HS and Hospedales, Timothy M},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2018}
}

References