pytorch-auto-drive icon indicating copy to clipboard operation
pytorch-auto-drive copied to clipboard

Changing the TuSimple output format to CULane format

Open AirSmoke031 opened this issue 3 years ago • 4 comments

Hello, I know this is not a part of your project, but I hit a deadend with my paper and I cant find the solution.

The papers is about taking 3 different datasets, in my case (Tusimple, CULane and LLAMAS) and using different models to test them, get the output and visualize. Then I have to use the same pretrained model (trained on TuSimple) and test it on the test set of CULane for example and vice versa.

The problem is the output format. LLAMAS (you already adjusted it to CULane format) and CULane use the .txt format so It shouldnt be a problem. However TuSimple has this specific format that they use with the .json file.

I want to know is there a way, a script, to reformat the .json TuSimple output format to that of CULane. If I had all the outputs in the same CULane format it would be easier for me to test a model trained on TuSimple and test it on CULane.

Sorry If I am nagging you, your repo is the main building block of my paper.

AirSmoke031 avatar Feb 22 '22 11:02 AirSmoke031

@AirSmoke031 If you load a tusimple trained pt file while using a CULane config pipeline (i.e. just modify the model so it can load the tusimple trained pt file). It should be able to inference. However, I would suggest using the same network arch and max lane num (recommend 4) for simplicity in your case. Also aligning input resolution seems good choice since lane detection is not sensitive to resolution.

voldemortX avatar Feb 22 '22 11:02 voldemortX

As for label format convertor, we don't have that yet.

voldemortX avatar Feb 22 '22 12:02 voldemortX

Thank you.

AirSmoke031 avatar Feb 22 '22 12:02 AirSmoke031

Anyone providing some kind of label conversion script, is encouraged to make a pull request and add them in /tools/

voldemortX avatar Mar 14 '22 01:03 voldemortX