dyabel / CoAR-ZSL

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CoAR-ZSL

This is the codes for the TNNLS paper Boosting Zero-shot Learning via Contrastive Optimization of Attribute Representations arxiv.

Prepare Data

Dataset: please download the dataset, i.e., CUB, AWA2, SUN, and put it under datasets/

Requirement

  • apex
  • PyTorch==1.7
  • timm

or you can build the environment by docker pull dythu/zsl

How to run

see train_cub.sh

About


Languages

Language:Python 99.6%Language:Shell 0.4%