界面已本地化;技能介绍与说明文档按所选语言由 AI 自动翻译,代码、命令、路径和专业术语保留原文,以原作者文档为准。
探索技能
数据与研究 · SKILL.md

data-preparation

data-preparation

【数据与研究】Prepare TuSimple labels, masks, index files, and TFRecords for LaneNet training.

独立 GitHub 快照 · 自动分类 · 以原作者文档为准

技能说明

GitHub

name: data-preparation description: "Prepare TuSimple labels, masks, index files, and TFRecords for LaneNet training." metadata: disco-role: operating disable-model-invocation: true license: Apache 2.0

Data preparation

Use this sub-skill to turn raw TuSimple labels into LaneNet-ready masks, dataset lists, and TFRecords.

Use when

  • You have an unzipped TuSimple archive and need training/gt_image, training/gt_binary_image, training/gt_instance_image, and training/train.txt.
  • You need a prepared dataset with gt_image/, gt_binary_image/, gt_instance_image/, train.txt, val.txt, test.txt, and tfrecords/tusimple_*.tfrecords.
  • You need to fix placeholder paths such as REPO_ROOT_PATH or ROOT_PATH before TFRecord generation.

Do not use for

  • Model fitting, checkpoint management, or loss/debugging. Use the training sub-skill.
  • Single-image inference, batch evaluation, or postprocess tuning. Use the inference-evaluation sub-skill.
  • TensorFlow checkpoint freezing or MNN export. Use the model-export sub-skill.

Start here

Bundled scripts

  • scripts/generate_tusimple_dataset.py converts raw TuSimple JSON labels into LaneNet training masks and a training/train.txt index.
  • scripts/make_tusimple_tfrecords.py validates the prepared dataset layout and writes tusimple_train.tfrecords, tusimple_val.tfrecords, and tusimple_test.tfrecords.

Expected outputs

  • Raw conversion: training/gt_image/, training/gt_binary_image/, training/gt_instance_image/, training/train.txt, and copied label JSONs under training/ and testing/.
  • TFRecord stage: tfrecords/tusimple_train.tfrecords, tfrecords/tusimple_val.tfrecords, tfrecords/tusimple_test.tfrecords.

Notes

  • The TFRecord wrapper auto-discovers the repository root and can override the dataset root in memory.
  • If the shipped sample lists still contain placeholders, regenerate or replace them before writing TFRecords.
我的收藏