【工具包使用】解析txt标注文件

发布时间 2023-11-06 18:40:50作者: 鹅要长大

前言

 

code

import os

def parse_onedata(data):
    print('data: ', data)
    info = data['info']
    image = data['image']
    valid = data['valid']
    rotation = data['rotation']
    categories = data['categories']
    annotations = data['annotations']
    print('image filename: ', image['file_name'])
    print('annotations len: ', len(annotations))

def parse_txtfile(path):
    imagedir = 'ID00011_female'
    filename = 'DCV20221108451508P0320221207114849.txt'
    filepath = os.path.join(path, imagedir, filename)
    print('imagedir: ', imagedir)
    txtfile = open(filepath)
    txtdata = txtfile.readlines()
    print('txtdata len: ', len(txtdata))
    for onedata in txtdata:
        print(type(onedata))
        onedata = onedata.replace('true', 'True')
        data = eval(onedata)
        print(type(eval(onedata)[0]))
        parse_onedata(data[0])
        break

if __name__ == "__main__":
    path = os.path.dirname(os.path.realpath(__file__))
    parse_txtfile(path)

 

参考

1. 数据开源 | Magic Data开源DMS驾驶员行为数据集

2. Multi-Modal Driver Behaviors Dataset for DMS - MagicHub