TRI-ML / KP3D

Code for "Self-Supervised 3D Keypoint Learning for Ego-motion Estimation"
MIT License
240 stars 29 forks source link

Train using our own data #13

Open MajdWardeh opened 2 years ago

MajdWardeh commented 2 years ago

Hi,

Thank you for this amazing paper! I was wondering if you could provide the code to train the models. I believe that the best way to prove the real performance of your approach is by letting people train it by themselves.

Thank you,

RaresAmbrus commented 2 years ago

Hi, thanks for the interest! Sadly we have no plans to release the training code, but happy to answer any specific questions about our method.

MajdWardeh commented 2 years ago

Hi @RaresAmbrus,

I am trying to implement the training part of KP3D. I have a couple of questions that I would be glad if you could help me with.

  1. After estimating the rotation, R, by solving the Orthogonal Procrestous problem, the resulting R could have a det of -1. What do you do to resolve this? do you implement Kabsch algorithm?
  2. For the geometric loss, It's stated that it is applied to the initial set of correspondences (the set obtained with descriptor matching). However, what about the outliers? is there any outlier rejection procedure or is there another rematching step (similar to KP2D)?
  3. For the descriptor loss, What are the positive descriptors associated with the anchor descriptors set? Are they defined by the inliers set coming from solving PnP with RANSAC? Otherwise, how could we find those positive descriptors?

Thank you so much (and for your colleagues) for this great paper! Kind regards, Majd Wardeh

MajdWardeh commented 2 years ago

Hello @RaresAmbrus,

I am sure you are very busy. I would be glad if you could help me clarify the questions above when you have the time.

Kind regards, Majd

Cc19245 commented 2 years ago

Hello @RaresAmbrus,

I am sure you are very busy. I would be glad if you could help me clarify the questions above when you have the time.

Kind regards, Majd

Hi, Majd! I'm also struggle to reproduce the training code of KP3D. Maybe we can get some communication? My email is Cherno19998@gmail.com. Looking forward to you reply!

RaresAmbrus commented 2 years ago

Hi, thanks for the interest and sorry about the slow reply, I'm out of the office and will only get back around the end of Nov. I'll try to take a look at the code. In the meantime, maybe @jiexiong2016 is available?