nuscenes数据集pkl文件转txt
nuscenes数据集pkl文件包含了数据集的重要信息,可以转成txt文件以方便查看。
pkl2txt.py
import pickle
import sys
pkl_name = "nuscenes_dbinfos.pkl" ## 可以直接修改文件名
if len(sys.argv)>1: ## 也可以通过参数方式输入pkl文件名
pkl_name = sys.argv[1]
# 打开PKL文件
with open(pkl_name, 'rb') as file:
data = pickle.load(file)
# 创建TXT文件
output_txt=pkl_name+'-output.txt'
with open(output_txt, 'w') as txt_file:
txt_file.write(str(data))
print("数据已保存到",output_txt,"文件中。")
可以直接修改文件名直接运行,也可以把文件名作为参数再运行。
python pkl2txt.py
或
python pkl2txt.py nuscenes_dbinfos2.pkl