tusimple数据集(json)转换脚本

最新推荐文章于 2025-06-25 10:51:42 发布

Kluiverthoo

最新推荐文章于 2025-06-25 10:51:42 发布

阅读量1w

点赞数 9

CC 4.0 BY-SA版权

分类专栏：深度学文章标签： python

本文链接：https://blog.youkuaiyun.com/derteanoo/article/details/82501337

本文介绍如何利用Python脚本将tusimple数据集从.json格式转换为Lanenet开源代码所需的图像格式，包括二值图、实例灰度图和原图，并提供数据集的下载链接及4:1比例划分训练集与测试集的脚本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Lanenet开源代码，用tusimple数据集进行训练，需要将.json数据集转换为readme中的图像格式。于是，手动写个python脚本转换工具。

lanenet github地址：https://github.com/MaybeShewill-CV/lanenet-lane-detection

tusimple数据集下载地址：http://benchmark.tusimple.ai/#/t/1/dataset

import cv2
import json
import numpy as np

file=open('label_data_0601.json','r')
image_num=0
for line in file.readlines():
    data=json.loads(line)
    image=cv2.imread(data['raw_file'])
    binaryimage=np.zeros((image.shape[0],image.shape[1],1),np.uint8)
    instanceimage=binaryimage.copy()
    arr_width=data['lanes']
    arr_height=data['h_samples']
    width_num=len(arr_width)
    height_num=len(arr_height)
    for i in range(height_num):
        lane_hist=0
        for j in range(width_num):
            if arr_width[j][i-1]>0 and arr_width[j][i]>