Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

SemanticTransfer

Code repo for the paper Semantic Correspondence via 2D-3D-2D Cycle.

Demo

Please run demo.py.

Pretrained Weights

You can download them from Google Drive.

Training

Training the full pipeline is somewhat involved and complicated, and our code is heavily based on ShapeHD. In general, there are four steps:

  • Train ShapeHD model as outlined here.
  • Prepare synthetic ShapeNet model renderings by mitsuba and generate their corresponding viewpoints through preprocess.py.
  • Train viewpoint estimation network by running scripts/train_vp.sh.
  • Train 3D embedding prediction network by running train_embs.py and then generate keypoints' average embeddings for retrieval. This step requires KeypointNet dataset.

About

Code repo for the paper "Semantic Correspondence via 2D-3D-2D Cycle"

Resources

Releases

No releases published

Packages

No packages published