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
839 stars 137 forks source link

about dataset #35

Closed miao02830 closed 2 years ago

miao02830 commented 2 years ago

In the code, I see that in the evaluation of the recognition results,use these two json files: ../../output/erfnet_baseline_tusimple.json & /home/data/Tusimple/test_label.json

I would like to ask you if I want to evaluate my dataset, do I also need to generate json files in this format. This file contains lanes, H samples,raw file. I've been looking for many websites these two days, but I haven't found a way to generate this JSON file.

If I want to use your method to train and evaluate my data set, is there any other method?

thankYou !!

voldemortX commented 2 years ago

There are mainly 2 formats for lane evaluation on the market. One is the TuSimple format as you just described, the other would be the CULane format. with a directory of txt files, maybe you can try if that format is easier? CULane metric does not constrain a fixed set of H_samples.

miao02830 commented 2 years ago

You mean that it is recommended to label my dataset in the format of culane instead of tusimple. It will be simpler. Is my understanding correct?thank you!

voldemortX commented 2 years ago

Well, it depends on what you have. Do you already have labels? Mainly, if they are labeled with fixed set of heights, TuSimple is better, if they are labeled with free points, CULane is better.

voldemortX commented 2 years ago

The CULane metric and annotation format is also better accepted by the community as well. You can see the new LLAMAS benchmark adopting this standard.

miao02830 commented 2 years ago

okay,i got it . i will try it.thankyou very much!

voldemortX commented 2 years ago

It seems the problem is resolved, I'll close for now. Feel free to reopen!