Update kitti_utils.py

HKUST-Aerial-Robotics · May 26, 2019 · 1641374 · 1641374
1 parent 19c465f
commit 1641374
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/model/utils/kitti_utils.py b/lib/model/utils/kitti_utils.py
@@ -441,6 +441,7 @@ def write_detection_results(result_dir, file_number, calib, box_left, pos, dim,
     '''One by one write detection results to KITTI format label files.
     '''
     if result_dir is None: return
+    result_dir = result_dir + '/data'
 
     # convert the object from cam2 to the cam0 frame
     dis_cam02 = calib.t_cam2_cam0[0]
@@ -453,7 +454,7 @@ def write_detection_results(result_dir, file_number, calib, box_left, pos, dim,
 
     # Write TXT files
     if not os.path.exists(result_dir):
-        os.mkdir(result_dir)
+        os.makedirs(result_dir)
     pred_filename = result_dir + '/' + file_number + '.txt'
     with open(pred_filename, 'a') as det_file:
         det_file.write(output_str)