sadmen
/
openpilot_comma
mirror of https://github.com/commaai/openpilot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

transformations: preserve dtype in denormalize

Browse Source
pull/22627/head
George Hotz 4 years ago
parent a7f36c9daf
commit cb01ef0919
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      common/transformations/camera.py

2
common/transformations/camera.py
Unescape View File

@ -131,7 +131,7 @@ def denormalize(img_pts, intrinsics=fcam_intrinsics, width=W, height=H):
img_pts = np.array(img_pts) img_pts = np.array(img_pts)
input_shape = img_pts.shape input_shape = img_pts.shape
img_pts = np.atleast_2d(img_pts) img_pts = np.atleast_2d(img_pts)
img_pts = np.hstack((img_pts, np.ones((img_pts.shape[0], 1)))) img_pts = np.hstack((img_pts, np.ones((img_pts.shape[0], 1), dtype=img_pts.dtype)))
img_pts_denormalized = img_pts.dot(intrinsics.T) img_pts_denormalized = img_pts.dot(intrinsics.T)
img_pts_denormalized[img_pts_denormalized[:, 0] > width] = np.nan img_pts_denormalized[img_pts_denormalized[:, 0] > width] = np.nan
img_pts_denormalized[img_pts_denormalized[:, 0] < 0] = np.nan img_pts_denormalized[img_pts_denormalized[:, 0] < 0] = np.nan

Write Preview
Loading…
Cancel
Save