提交保存文件

script/trans_data.py

@@ -484,7 +484,7 @@ if __name__ == '__main__':
     for file_path,boxx in anno.items():
       svaeFile = save_root_path + saveName[file_path] + ".csv"
       f1 = open(svaeFile, 'w')
-      f1.write('lat,lon,x,y,z,l,w,h,rot_y')
+      f1.write('lat,lon,x,y,z,l,w,h,rot_y\n')
       # print(file_path)
       # print(boxx)
       bbox_flag,bboxes = read_all_3d_bboxes(boxx, np.identity(4))
@@ -506,9 +506,9 @@ if __name__ == '__main__':
         lidar_loc, out_BL, out_center_BL = get_loc([bbox[4][0], bbox[4][1], bbox[4][2], bbox[6][0], bbox[6][1], bbox[6][2], bbox[2]], 0, origin_Trans, origin_kitti2origin)
         print(out_center_BL)
         angle = bbox[2]/np.pi*180
-        isInMap = get_map_data(out_center_BL[0],out_center_BL[1],angle)
-        print("call get data isInMap = ",isInMap)
-
+        mapInfo = get_map_data(out_center_BL[0],out_center_BL[1],angle)
+        print("call get data isInMap = ",mapInfo)
+        f1.write('%f,%f,%f,%f,%f,%f,%f,%f,%f\n'%(out_center_BL[0],out_center_BL[1],bbox[4][0], bbox[4][1], bbox[4][2],bbox[6][0], bbox[6][1], bbox[6][2],angle))
         flag = in_hull(xyz,bbox[1])
         pickcloud = xyz[flag]
         dz = np.array([0,0,0])