diff --git a/lib/model/utils/kitti_utils.py b/lib/model/utils/kitti_utils.py index 09bd90c..4ba8a9e 100644 --- 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)