search

koyeongmin / PINet_new

MIT License
166 stars 44 forks source link

Translation and Rotate #10

Closed jiayily closed 3 years ago

jiayily commented 4 years ago

Hi, I have some question about Translation and Rotate in data_loader.py. https://github.com/koyeongmin/PINet_new/blob/fead6baf0210a11444bf02d8fda7b3a33cec8e8c/TuSimple/data_loader.py#L384 Why don't you process target_h when you translate and rotate the image in case that target_h is out of [0, y_size-1]?

koyeongmin commented 3 years ago

Because y_size is height of input image, It will occur error if we make some value at that point.

jiayily commented 3 years ago

Probably I didn't make myself clear to you. When you translate and rotate the image, why don't you translate and rotate the target_h that is y coordinates of lane keypoints? In your code, you only translate and rotate target_lanes that is x coordinates of lane keypoints.