423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 20:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix docstring of mesh edge loss

Browse Source 
Summary: Fix docstring for mesh edge loss

Reviewed By: jcjohnson

Differential Revision: D20040560

fbshipit-source-id: 01a3ee9473c7d11583684bf4cd200caa1d3f0260
This commit is contained in:
Georgia Gkioxari 2020-02-21 15:54:14 -08:00 committed by Facebook Github Bot
parent e491efb81f
commit c2a0a3e3ba
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/loss/mesh_edge_loss.py
View File

@ -7,8 +7,8 @@ import torch
def mesh_edge_loss(meshes, target_length: float = 0.0): def mesh_edge_loss(meshes, target_length: float = 0.0):
""" """
Computes mesh edge length regularization loss averaged across all meshes Computes mesh edge length regularization loss averaged across all meshes
in a batch. Each edge contributes equally to the final loss, regardless of in a batch. Each mesh contributes equally to the final loss, regardless of
numbers of edges per mesh in the batch by weighting each mesh with the the number of edges per mesh in the batch by weighting each mesh with the
inverse number of edges. For example, if mesh 3 (out of N) has only E=4 inverse number of edges. For example, if mesh 3 (out of N) has only E=4
edges, then the loss for each edge in mesh 3 should be multiplied by 1/E to edges, then the loss for each edge in mesh 3 should be multiplied by 1/E to
contribute to the final loss. contribute to the final loss.