Sketch-based 3D Shape Retrieval

The code is used in the paper of JMLN, which attempts to further improve the cross-domain retrieval task. The framework is shown as followings:

The code has been tested successfully on Ubuntu 18.04.3 LTS, Cuda 10.0, GTX 1080 Ti, Pytorch 1.4.0, Docker images ufoym/deepo:all-py36-cu100.

Note: Using docker is suggested.

Training and Test Data

Please put related dataset into dataset/

# training 
sh train_all.sh

# Test after training
sh extract_all.sh