Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train my own dataset? #55

Closed
NARRZ opened this issue Dec 13, 2021 · 3 comments
Closed

How to train my own dataset? #55

NARRZ opened this issue Dec 13, 2021 · 3 comments

Comments

@NARRZ
Copy link

NARRZ commented Dec 13, 2021

Hello, I am very interested in your research results. My own lane line data set has been converted to tusimple style. How can I use my own data set for training? Thank you!

@lucastabelini
Copy link
Owner

lucastabelini commented Dec 13, 2021

Thank you for your interest. You might also be interested in my latest lane detection work: https://github.com/lucastabelini/LaneATT

If your dataset is in TuSimple's format, you can train a model on it using the same procedure described on the README.md: https://github.com/lucastabelini/PolyLaneNet#training

One of the settings that you will have to change is the root of the dataset (which points to "datasets/root" by default). You will also want to modify the number of epochs and the learning rate scheduler.

@NARRZ
Copy link
Author

NARRZ commented Dec 15, 2021

Thank you for your reply. I found that I did not really convert my dataset to tusimple format, and I did not find the correct conversion method, do you have any suggestions to convert the data set to tusimple format? Thank you!

@lucastabelini
Copy link
Owner

See #5.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants