对KITTI -360数据集 data_3d_raw 002 文件夹批量重命名
import argparse
import h5py
import open3d as o3d
import os
import torch
from tqdm import tqdm
import numpy as np
import struct
import os
parser = argparse.ArgumentParser()
parser.add_argument(‘–root_folder’, default=‘/media/oem/G/KITTI360’, help=‘dataset directory’)
args = parser.parse_args()
base_dir = args.root_folder
out_dir = ‘data2’
def WriteFile():
for sequence in [“2013_05_28_drive_0002_sync”]:
file_list = os.listdir(os.path.join(base_dir, ‘data_3d_raw’, sequence, ‘velodyne_points’, ‘data’))
for idx in range((len(file_list))):
idx += 4391
global velo_path
velo_path = os.path.join(base_dir, ‘data_3d_raw’, sequence, ‘velodyne_points’, ‘data’,f’{idx:010d}.bin’)
# 设置旧文件名(就是路径+文件名)
oldname = velo_path # os.sep添加系统分隔符
# 设置新文件名
idx -=4391
save_path = os.path.join(base_dir, 'data_3d_raw', sequence, 'velodyne_points', 'data', f'{idx:010d}.bin')
newname = save_path
os.rename(oldname, newname) # 用os模块中的rename方法对文件改名
print(oldname, '======>', newname)
#save_file = os.path.join(base_dir, 'data_3d_raw', sequence, 'velodyne_points', out_dir, f'{idx:010d}.bin')
if name == ‘main’:
WriteFile()