AMFF

AMFF

Feature Fusion and Metric Learning Network for Zero-Shot Sketch-Based Image Retrieval

Website: https://www.mdpi.com/1099-4300/25/3/502
PDF Version: https://www.mdpi.com/1099-4300/25/3/502/pdf

#setup

1.ubuntu >= 20.04 LTS

2.conda create -n AMFF python=3.8

3.conda activate AMFF

4.pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

5.python setup.py install

6.pip install pandas pytorch_metric_learning timn scikiy-image

7.pip install pretrainedmodels nltk

Data set directory structure

-data/

-sketchy/

-train/

-photo/

-image1

-sketch/

-image1

-val/

-photo/

-image1

-sketch/

-image1

Train Model

para

--proj_dim 512

--backbone_type resnet50 --data_name sketchy

--tri_lambda 1 --batch_size 64

--num-workers 8 --loss mathm --epochs 10

dataset

sketchy and tu-berlin

cite

If our code helps you, please cite

@article{zhao2023feature,
title={Feature Fusion and Metric Learning Network for Zero-Shot Sketch-Based Image Retrieval},
author={Zhao, Honggang and Liu, Mingyue and Li, Mingyong},
journal={Entropy},
volume={25},
number={3},
pages={502},
year={2023},
publisher={MDPI}
}