diff --git a/dgp/utils/structures/key_point_3d.py b/dgp/utils/structures/key_point_3d.py index b2ce80c8..3a03c5df 100644 --- a/dgp/utils/structures/key_point_3d.py +++ b/dgp/utils/structures/key_point_3d.py @@ -91,4 +91,4 @@ def to_proto(self): KeyPoint3D.pb2 As defined in `proto/annotations.proto` """ - return annotations_pb2.KeyPoint3D(x=int(self.x), y=int(self.y), z=int(self.z)) + return annotations_pb2.KeyPoint3D(x=float(self.x), y=float(self.y), z=float(self.z))