Skip to content

yitao416/local3Ddescriptorlearning

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 

Repository files navigation

local3Ddescriptorlearning

This code implements the method of deep learning local descriptors for 3D surface shapes described in the following paper:

  --------------------------------------------------------------
  Hanyu Wang, Jianwei Guo, Dong-Ming Yan, Weize Quan, Xiaopeng Zhang. 
  Learning 3D Keypoint Descriptors for Non-Rigid Shape Matching. 
  ECCV 2018.
  --------------------------------------------------------------

Please consider citing the above paper if you use the code/program (or part of it).

License

This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version.

Usage

There are three folders here. The "cpp" calls the "matlab" for GI generation. The "python" is used for network training and testing.

The usage is as follows:

  1. Compile matlab project:

     MCC matlab "mcc -W cpplib:libcompcur -T link:lib compute_curvature.m". We got the libcompcur.dll that will be added to the CPP project.
    
  2. Build cpp solution: this code is to generate geometry images. You can run this step in your local desktop.

     Modify CMakeLists :
     	add include_directories and link_directories for openmesh and matlab runtime
     
     Cmake
     
     Build solution
     
     Modify config.ini for mesh_dir(directory of OFF models) gi_dir(directory of geometry images) and kpi_dir(directory of key points, you can skip it for dense matching)
     	edit other paras such as gi_size(NxN of gi), hks_len, rotation_num and radius_list_p (the ratio of geodesic diameter).
     
     Add "libcompcur.dll" to folder with GIGen.exe
     
     Run "GIGen.exe config.ini" to generate GI
    
  3. Python project: this code is to train and test network. You should copy the geometry images generated in last step into the server.

    -Train network:

     run "classify_gi_by_pidx_and_split.py" to classify dataset by point index for generating Tfrecord
     	source_dir is the folder of geometry images (rotation_num=12) for training, destination_dir is the generated folder after splitting
     
     run "tfr_gen.py" to generate Tfrecord
     	gi_dir is the same as destination_dir, tfr_dir is the generated folder of Tfrecord.
     
     run "train_softmax256.py" to pretrain a classification network
     	tfr_dir needs to be specified
     
     run "train_mincv_perloss.py" to train the triplet network by restoring a pre-trained classification model
     	Tfr_dir needs to be specified
    

    -Test to generate descriptor:

     run "descGen.py" to generate descriptor using test dataset
     	gi_dir is the folder of geometry images (rotation_num=1) for testing, desc_dir is the generated folder of descriptor.
    

Contact

Should you have any questions, comments, or suggestions, please contact me at: [email protected]

Jianwei Guo: http://jianweiguo.net/

October, 2018

Copyright (C) 2018

About

Deep learning local descriptors for 3D surface shapes

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 57.5%
  • Python 24.5%
  • C 12.2%
  • MATLAB 5.2%
  • CMake 0.6%