voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
BSD 3-Clause "New" or "Revised" License
850 stars 138 forks source link

tusimple視覺可視化問題 #82

Closed j2856256 closed 2 years ago

j2856256 commented 2 years ago

您好 想請問一下 如何將tusimple test完後json檔修改為culane的格式 tusimple json 檔,內含h_samples、lanes、run_time、raw_file,若要可視化是不是只能改成像測試檔案一樣,單一圖片只有lanes的json檔

j2856256 commented 2 years ago

Screenshot from 2022-04-28 20-54-04 tusimple json檔

j2856256 commented 2 years ago

Screenshot from 2022-04-28 20-55-54 測試檔案

voldemortX commented 2 years ago

@j2856256 I‘m assuming you speak Chinese.

你是要可视化Tusimple数据集还是你有其他数据集用了这个格式?

如果是TuSimple数据集本身是有相关的特殊支持的。虽然写成了编程教程,但是本身已经包括在了master分支: https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/advanced/VISUALIZE_LANE_DATASETS.md

如果是其他数据集用了类似的标注,或者只是普通数据输入没有数据集,你可以参照这个教程做有针对性地实现。

如果是真的要做格式转换,那现在没有这个功能,但我们很希望有人能贡献这个功能。#59

j2856256 commented 2 years ago

utils/datasets/tusimple_vis.py具體要如何使用可以提供具體的執行碼嗎?

voldemortX commented 2 years ago

utils/datasets/tusimple_vis.py具體要如何使用可以提供具體的執行碼嗎?

首先需要在config文件增加vis_dataset,然后才可以运行。请仔细阅读这一段: https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/advanced/VISUALIZE_LANE_DATASETS.md#expected-usage

j2856256 commented 2 years ago

添加之後就可以了,非常感謝 那我可以再問一個問題嗎? 如何讓tusimple test全部都有輸出,因為該作法只有第20個fps有結果

j2856256 commented 2 years ago

如何讓tusimple 全部test資料都有輸出,因為該作法只有第20個fps有結果

voldemortX commented 2 years ago

如何讓tusimple 全部test資料都有輸出,因為該作法只有第20個fps有結果

你指的是tusimple测试时产生的json文件么,那里面的fps都是不做测试的。速度测试有专用的脚本,与评测过程不同的是分割等方法对齐到tusimple格式的后处理是不计入速度测试的:

https://github.com/voldemortX/pytorch-auto-drive/blob/master/docs/BENCHMARK.md#profiling-models-yourself

voldemortX commented 2 years ago

The issue seems to be resolved. I'll close for now. Feel free to ask for reopen or open new issues.

本issue已基本解决,暂时关闭。如果有进一步的疑惑,可以随时reopen或者开新的。