Mercurial Hosting > traffic-intelligence
diff trafficintelligence/storage.py @ 1233:d5695e0b59d9
saving results from ultralytics works
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 08 Sep 2023 17:09:12 -0400 |
parents | 83ca1493d55c |
children | 31173c4699d2 |
line wrap: on
line diff
--- a/trafficintelligence/storage.py Thu Sep 07 16:20:28 2023 -0400 +++ b/trafficintelligence/storage.py Fri Sep 08 17:09:12 2023 -0400 @@ -10,7 +10,7 @@ from numpy import log, min as npmin, max as npmax, round as npround, array, sum as npsum, loadtxt, floor as npfloor, ceil as npceil, linalg, int32, int64, reshape, dot, vstack, transpose, ones, zeros_like, pi, NaN from pandas import read_csv, merge, concat, DataFrame -from trafficintelligence import utils, moving, events, indicators, cvutils +from trafficintelligence import utils, moving, events, indicators from trafficintelligence.base import VideoFilenameAddable @@ -1307,6 +1307,7 @@ Ref: https://github.com/pratikac/kitti/blob/master/readme.tracking.txt''' from pykitti.utils import roty + from trafficintelligence.cvutils import cartesian2Homogeneous invR0 = linalg.inv(reshape(kittiCalibration['R_rect'], (3, 3))) transCam2Velo = transpose(kittiCalibration['Tr_cam_velo']) @@ -1371,16 +1372,16 @@ # corners3d = transpose(corners3d) # box3d_pts_3d_velo = calib.project_rect_to_velo(box3d_pts_3d) refCorners = transpose(dot(invR0, corners3d)) # 8x3 avoid double transpose np.transpose(pts_3d_rect))) in pts_3d_ref = self.project_rect_to_ref(pts_3d_rect) - homRefCorners = cvutils.cartesian2Homogeneous(refCorners) + homRefCorners = cartesian2Homogeneous(refCorners) veloCorners = dot(homRefCorners, transCam2Velo) # self.project_ref_to_velo(pts_3d_ref) # boxes3d_world_single, flag_imu, flag_world = calib_function.project_velo_to_world(bboxes_3d_velo=boxes3d_velo_single, Tr_matrix=transfer_matrix) #Tr_velo_to_imu = transfer_matrix['Tr_velo_to_imu'] #Tr_imu_to_world = transfer_matrix['Tr_imu_to_world'] - homVeloCorners = cvutils.cartesian2Homogeneous(veloCorners) + homVeloCorners = cartesian2Homogeneous(veloCorners) imuCorners = dot(kittiCalibration['Tr_velo_imu'], homVeloCorners.T).T # 8x3 - homImuCorners = cvutils.cartesian2Homogeneous(imuCorners) + homImuCorners = cartesian2Homogeneous(imuCorners) worldCorners = dot(Tr_imu_to_world, homImuCorners.T).T # 8x3 # take first 4 lines of corners, x,y,_ # x0, y0, _ = boxes3d[0] @@ -1415,12 +1416,12 @@ corners3d[1, :] = corners3d[1, :] corners3d[2, :] = corners3d[2, :] refCorners = transpose(dot(invR0, corners3d)) # 8x3 avoid double transpose np.transpose(pts_3d_rect))) in pts_3d_ref = self.project_rect_to_ref(pts_3d_rect) - homRefCorners = cvutils.cartesian2Homogeneous(refCorners) + homRefCorners = cartesian2Homogeneous(refCorners) veloCorners = dot(homRefCorners, transCam2Velo) - homVeloCorners = cvutils.cartesian2Homogeneous(veloCorners) + homVeloCorners = cartesian2Homogeneous(veloCorners) imuCorners = dot(kittiCalibration['Tr_velo_imu'], homVeloCorners.T).T # 8x3 - homImuCorners = cvutils.cartesian2Homogeneous(imuCorners) + homImuCorners = cartesian2Homogeneous(imuCorners) worldCorners = dot(Tr_imu_to_world, homImuCorners.T).T # 8x3 # take first 4 lines of corners, x,y,_ # x0, y0, _ = boxes3d[0]