Andy97 / DeepMLS

Deep Implicit Moving Least-Squares Functions for 3D Reconstruction
MIT License
130 stars 22 forks source link

TFRecords preprocessing #3

Closed bluestyle97 closed 3 years ago

bluestyle97 commented 3 years ago

Hi, I want to generate tfrecords training data from another mesh dataset (which has already been watertight), but there seems to be no instruction on this step. I have read this istruction, but the code seems to compute sdf values only and have no help with my question. I found there are fields in the tfrecords data that I don't know what they means and where they come from, such as "sdf_samples_depth6", "sdf_samples_depth7", "depth6_coord" and "depth7_coord". Can you give a more detailed instruction on how to generate the right-format tfrecords data? Thanks a lot!

Andy97 commented 3 years ago

Hi, I want to generate tfrecords training data from another mesh dataset (which has already been watertight), but there seems to be no instruction on this step. I have read this istruction, but the code seems to compute sdf values only and have no help with my question. I found there are fields in the tfrecords data that I don't know what they means and where they come from, such as "sdf_samples_depth6", "sdf_samples_depth7", "depth6_coord" and "depth7_coord". Can you give a more detailed instruction on how to generate the right-format tfrecords data? Thanks a lot!

Hi, as mentioned in the paper, we have 2 set of sdf samples(sampled from 64x64x64 grid points and 128x128x128 grid points, please refer to Appendix A of paper), which were used for sdf training in a coarse-to-fine manner progressively(e.g. using 64^3 samples for first 30 epochs and 128^3 samples for remaining epochs). The "depth6_coord" and "depth7_coord" is the 3d coordinates of these sample points, while "sdf_samples_depth6", "sdf_samples_depth7" contain the corresponding ground truth sdf values and gradients. For details, please refer to code for data loading.

For memory efficiency, each 3-dimensional vector "depth6_coord" and "depth7_coord" is compressed to an int(from 0x0x0 to 257x257x257). (Please refer to here).

The field "octree_points_idx" contains the ground truth octree shape, which has also been compressed to an integer.

Andy97 commented 3 years ago

Script for generating tfrecords is released, please see here.