adapter new type promotion rule for Paddle 2.6 (#455)

* adapter new type promotion rule for Paddle 2.6 * fix code style
PaddlePaddle · Mar 11, 2024 · 3259dab · 3259dab
1 parent 1d846ee
commit 3259dab
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/paddle3d/models/detection/smoke/smoke_coder.py b/paddle3d/models/detection/smoke/smoke_coder.py
@@ -133,9 +133,8 @@ def decode_location(self, points, points_offset, depths, Ks, trans_mats):
         points = paddle.reshape(points, (-1, 2))
         assert points.shape[0] == N
 
-        # int + float -> int, but float + int -> float
         # proj_points = points + points_offset
-        proj_points = points_offset + points
+        proj_points = points_offset + points.astype(points_offset.dtype)
 
         # transform project points in homogeneous form.
         proj_points_extend = paddle.concat(