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

Alternative type names in PLY #205

Browse Source 
Summary: Add ability to decode ply files which use types like int32.

Reviewed By: nikhilaravi

Differential Revision: D21639208

fbshipit-source-id: 0ede7d4aa353a6e940446680a18e7ac0c48fafee
This commit is contained in:
Jeremy Reizenstein 2020-05-20 01:11:39 -07:00 committed by Facebook GitHub Bot
parent b4fd9d1d34
commit f2d1d2db69
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pytorch3d/io/ply_io.py
View File

@ -26,6 +26,14 @@ _PLY_TYPES = {
"uint": _PlyTypeData(4, "I", np.uint32),
"float": _PlyTypeData(4, "f", np.float32),
"double": _PlyTypeData(8, "d", np.float64),
"int8": _PlyTypeData(1, "b", np.byte),
"uint8": _PlyTypeData(1, "B", np.ubyte),
"int16": _PlyTypeData(2, "h", np.short),
"uint16": _PlyTypeData(2, "H", np.ushort),
"int32": _PlyTypeData(4, "i", np.int32),
"uint32": _PlyTypeData(4, "I", np.uint32),
"float32": _PlyTypeData(4, "f", np.float32),
"float64": _PlyTypeData(8, "d", np.float64),
}
_Property = namedtuple("_Property", "name data_type list_size_type")
@ -84,9 +92,9 @@ class _PlyElementType:
"""
if not self.is_fixed_size():
return False
first_type = self.properties[0].data_type
first_type = _PLY_TYPES[self.properties[0].data_type]
for property in self.properties:
if property.data_type != first_type:
if _PLY_TYPES[property.data_type] != first_type:
return False
return True

2
tests/test_ply_io.py
View File

@ -284,7 +284,7 @@ class TestMeshPlyIO(TestCaseMixin, unittest.TestCase):
format,
"element vertex 8",
"property float x",
"property float y",
"property float32 y",
"property float z",
"element vertex1 8",
"property float x",