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

Pointclouds, Meshes and Textures self-references

Browse Source 
Summary: Use `self.__class__` when creating new instances, to slightly accommodate inheritance.

Reviewed By: nikhilaravi

Differential Revision: D21504476

fbshipit-source-id: b4600d15462fc1985da95a4cf761c7d794cfb0bb
This commit is contained in:
Jeremy Reizenstein
2020-05-11 12:55:30 -07:00
committed by Facebook GitHub Bot
parent 34a0df0630
commit 6a365d203f
3 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pytorch3d/structures/meshes.py
View File

@@ -436,9 +436,9 @@ class Meshes(object):
textures = None if self.textures is None else self.textures[index]
if torch.is_tensor(verts) and torch.is_tensor(faces):
return Meshes(verts=[verts], faces=[faces], textures=textures)
return self.__class__(verts=[verts], faces=[faces], textures=textures)
elif isinstance(verts, list) and isinstance(faces, list):
return Meshes(verts=verts, faces=faces, textures=textures)
return self.__class__(verts=verts, faces=faces, textures=textures)
else:
raise ValueError("(verts, faces) not defined correctly")
@@ -1127,7 +1127,7 @@ class Meshes(object):
faces_list = self.faces_list()
new_verts_list = [v.clone() for v in verts_list]
new_faces_list = [f.clone() for f in faces_list]
other = Meshes(verts=new_verts_list, faces=new_faces_list)
other = self.__class__(verts=new_verts_list, faces=new_faces_list)
for k in self._INTERNAL_TENSORS:
v = getattr(self, k)
if torch.is_tensor(v):
@@ -1370,7 +1370,7 @@ class Meshes(object):
if self.textures is not None:
tex = self.textures.extend(N)
return Meshes(verts=new_verts_list, faces=new_faces_list, textures=tex)
return self.__class__(verts=new_verts_list, faces=new_faces_list, textures=tex)
def join_meshes_as_batch(meshes: List[Meshes], include_textures: bool = True):