【无标题】

对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()