search

vislearn / ngdsac_horizon

Neural-Guided, Differentiable RANSAC for Horizon Line Estimation
BSD 3-Clause "New" or "Revised" License
33 stars 4 forks source link

how to create local data for training and what is the training label format? #5

Open learn01one opened 5 months ago

learn01one commented 5 months ago

Hello, how to create local data for training and what is the training label format? , the hlw data cannot be downloaded and the data format cannot be known, thank you

ebrach commented 3 months ago

Hi. Sorry to hear that the HLW dataset is not accessible anymore. The request form (linked in our readme) still seems to exist - did you request access?

Anyway, another option to infer the dataset format is to reverse engineer hlw_dataset.py, specifically looking at https://github.com/vislearn/ngdsac_horizon/blob/1bf0172f5094101f750d87b7549643e25c3c17b9/hlw_dataset.py#L48-L59 and https://github.com/vislearn/ngdsac_horizon/blob/1bf0172f5094101f750d87b7549643e25c3c17b9/hlw_dataset.py#L73-L84.

It looks to me like the metadata.csv of the dataset contained one line per image, as img_file, x0, y0, x1, y1. The line coordinates (start and end point) seem to be in px, and relative to the image center. y-direction seems to be from bottom to top.

But all of the above is a guess based on a quick look and has to be tested.

Best, Eric