Skip to content

Commit

Permalink
fix EMD readme
Browse files Browse the repository at this point in the history
  • Loading branch information
Colin97 committed Dec 21, 2019
1 parent ee85ec0 commit e4fcb52
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions emd/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@ See `emd_module.py/test_emd()` for examples.
### Input

- **xyz1, xyz2**: float tensors with shape `[#batch, #points, 3]`. xyz1 is the predicted point cloud and xyz2 is the ground truth point cloud. Two point clouds should have same size and be normalized to [0, 1]. The number of points should be a multiple of 1024. The batch size should be no greater than 512. Since we only calculate gradients for xyz1, please do not swap xyz1 and xyz2.
- **eps**: a float tensor. The parameter balances the error rate and the speed of convergence.
- **iters**: a int tensor. The number of iterations.
- **eps**: a float tensor, the parameter balances the error rate and the speed of convergence.
- **iters**: a int tensor, the number of iterations.

### Output

- **dist**: a float tensor with shape `[#batch, #points]`. sqrt(dist) are the L2 distance between the pairs of points.
- **dist**: a float tensor with shape `[#batch, #points]`. sqrt(dist) are the L2 distances between the pairs of points.
- **assignment**: a int tensor with shape `[#batch, #points]`. The index of the matched point in the ground truth point cloud.

0 comments on commit e4fcb52

Please sign in to comment.