提交更新

script/trans_and_export_data_multi.py

@@ -479,6 +479,7 @@ def Save_Cloud_File(cloud,json):
     jsn_pcd = {}
     
     jsn_pcd["labels_box3D"] = []
+    jsn_pcd["box_num"] = {'big': 0, 'little':0,'pedestrian':0,'mid':0,'cyclist':0}
     for bbox in cloud["boxes"]:
         jsn_box = {}
         jsn_box["box3D"] = {}
@@ -487,6 +488,7 @@ def Save_Cloud_File(cloud,json):
         jsn_box["box3D"]["rot_y"] = bbox[2]
         jsn_box["category"] = bbox[0]
         jsn_pcd["labels_box3D"].append(jsn_box)
+        jsn_pcd["box_num"][bbox[0]] += 1
         
         flag_del = shadow_casting(cloud["np_pcd"],bbox[1])
         #g_np_flag = g_np_flag & (~flag_del)