txt2npy

最新推荐文章于 2024-09-02 20:01:37 发布

原创最新推荐文章于 2024-09-02 20:01:37 发布 · 2.2k 阅读

2 ·

CC 4.0 BY-SA版权

点云深度学习相关专栏收录该内容

12 篇文章

订阅专栏

txt2npy

import os
import sys
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
ROOT_DIR = os.path.dirname(BASE_DIR)
sys.path.append(BASE_DIR)
import numpy as np

#anno_paths = [line.rstrip() for line in open(os.path.join(BASE_DIR, 'meta/anno_paths.txt'))]
#anno_paths = [os.path.join(indoor3d_util.DATA_PATH, p) for p in anno_paths]

data_filelists = [line.rstrip() for line in open('/media/david/8TB数据盘/VITTK/pointnet(add_model)/data/KITTI_RAW_DATA/file_list.txt')]

output_folder = os.path.join(ROOT_DIR, 'data/KITTI_RAW_DATA/npy') 
if not os.path.exists(output_folder):
    os.mkdir(output_folder)

# Note: there is an extra character in the v1.2 data in Area_5/hallway_6. It's fixed manually.
for anno_path in data_filelists:
    print(anno_path)

    elements = anno_path.split('.')
    out_filename = elements[-2] + '.npy' 
    data = np.loadtxt(anno_path)
    data[...,-1].astype(int)
    np.save(out_filename, data)