Skip to content

Latest commit

 

History

History
29 lines (16 loc) · 633 Bytes

README.md

File metadata and controls

29 lines (16 loc) · 633 Bytes
Raw

DRCPL

Deep Residual Coupled Prompt Learning for Zero-Shot Sketch-Based Image Retrieval

Fig.1

========= Installation and Requirements =========

  • cudatoolkit=11.3.1

  • numpy=1.19.3

  • python=3.7.16

  • pytorch=1.10.0

  • torchvision=0.11.0

============== Datasets ==============

Sketchy and TU-Berlin

Please go to the ZSE-SBIR

============== Training ==============

  • CUDA_VISIBLE_DEVICES=1 python train.py

============== Testing ==============

  • CUDA_VISIBLE_DEVICES=1 python test.py