/CRGN

Deep Relation Reasoning for Cross-Modal Retrieval

Primary LanguagePython

CRGN

Deep Relation Embedding for Cross-Modal Retrieval

This code is implemented by pytorch.
The meaning of file and dir:
final_CRGN and final_CRGN_f30k is related to MSCOCO and Flickr30K datasets, respectively.
runs:Save generated model and log
data.py:Process data
evaluation.py,evaluation_zyf.py:Evaluating model performance on Recall
model.py,model_zyf.py:Network structure
test.py:Test model
train.py:Train model
vocab.py:Generate vocabulary

How to train:python train.py --max_violation
How to test:python test.py
To reproduce the results in the paper, please follow the training procedure in the paper.