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

Creating a {descriptor, class} object dataset #8

Closed
nikhilbv opened this issue May 18, 2020 · 6 comments
Closed

Creating a {descriptor, class} object dataset #8

nikhilbv opened this issue May 18, 2020 · 6 comments

Comments

@nikhilbv
Copy link

Hi Fabvio,

Great work. I would love to know when the training code will be released. Meanwhile, I'm trying to train only the classifier CNN and for that, I'm trying to create a {descriptor, class} dataset as mentioned in the paper. In my approach, I'm directly drawing the lanes on an image(binary) from the tusimple dataset lane annotations, resizing the original and binary image to 360*640, converting it to torch tensors, and passing it to extract_descriptor function.

Expected Output -
For a four-lane image, four descriptors should be extracted

Actual output -
No descriptor is extracted

I am wondering if my approach is correct and if it is correct, then why is the extract_descriptor function is not working as expected.

@fabvio
Copy link
Owner

fabvio commented May 18, 2020

Hi. I guess that the function is not working because you are passing a binary image! Check the jupyter notebook and you'll see that you need an image with instance labels projected on in order to extract the descriptors :)

@nikhilbv
Copy link
Author

nikhilbv commented May 19, 2020

Hi! Thanks for the reply. Does the width of the lane matters in the image with instance labels projected? I have made an instance image using tusimple dataset annotations. It looks like this.
0.
I've used the width of 5px for drawing polyline but still not able to extract descriptors

@fabvio
Copy link
Owner

fabvio commented May 19, 2020

This is the actual image you are passing to the function? Because you shouldn't pass a color image, you should pass a greyscale one with contiguous lane ids, e.g. 1,2,3,4 instead of (0,255,0),(255,0,0) etc.

@fabvio
Copy link
Owner

fabvio commented May 24, 2020

Closed for inactivity. Feel free to reopen if there is the need.

@fabvio fabvio closed this as completed May 24, 2020
@jiguanglu
Copy link

trying to train only the classifier CNN
Hello, can you share the code for the training classifier CNN? I will be very grateful to you.

@jiguanglu
Copy link

This is the actual image you are passing to the function? Because you shouldn't pass a color image, you should pass a greyscale one with contiguous lane ids, e.g. 1,2,3,4 instead of (0,255,0),(255,0,0) etc.

Hello, can you share the code for the training classifier CNN? I will be very grateful to you.

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

3 participants