tarun738 / i3DMM

Other
84 stars 9 forks source link

Hi, I got a bad reconstruction result of mesh from the reference network #5

Open yiyuzhuang opened 3 years ago

yiyuzhuang commented 3 years ago

I selected the "10084_e17" model from the test set you provided, and tried to use it to pretrain the reference network as your paper. However, when I try to reconstruct the training result, I found it's terribly bad. Could you please tell me what's wrong with it? Here is my result: image and there are lots of holes on it image

And the ground truth: image

I use these three command:

  1. prepare the training data set python preprocessData.py --samples_directory ./data --input_meshes_directory ../dataset -e headModel -s Train
  2. train the model python train_i3DMM.py -e headModel -s ./headModel/splits/referenceShapeSample.json -d data -r 1
  3. reconstruct the reference network result python fit_i3DMM_to_mesh.py -e headModel -s ./headModel/splits/referenceShapeSample.json -d data -r 1
tarun738 commented 2 years ago

Hello!

Thank you for your interest in our paper!

This is expected since the data is noisy. You need to train the entire model and not just the reference when over fitting. Also, since you're over fitting to a mesh, you need to train for more epochs. (Around 5-10k, generally takes an hour, see you the other issue that was just closed)

We only use the trained reference as an initialization. We also learn the reference while training the entire network.

Hope that answers.

Best, Tarun