Python 3.6.7
PyTorch = 1.7.0
../../datasets/
.extract_attribute_w2v_CUB.py
extract_attribute_w2v_SUN.py
extract_attribute_w2v_AWA2.py
should generate and place it in w2v/
.Before running commands, you can set the hyperparameters in config on different datasets:
config/cub.yaml #CUB
config/sun.yaml #SUN
config/awa2.yaml #AWA2
T rain:
python train.py
Eval:
python test.py
You can test our trained model: CUB, AwA2, SUN.
If this work is helpful for you, please cite our paper.
@InProceedings{Liu_2023_CVPR,
author = {Liu, Man and Li, Feng and Zhang, Chunjie and Wei, Yunchao and Bai, Huihui and Zhao, Yao},
title = {Progressive Semantic-Visual Mutual Adaption for Generalized Zero-Shot Learning},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2023},
pages = {15337-15346}
}
We thank the following repos providing helpful components in our work. GEM-ZSL