search

lucastabelini / PolyLaneNet

Code for the paper entitled "PolyLaneNet: Lane Estimation via Deep Polynomial Regression" (ICPR 2020)
https://arxiv.org/abs/2004.10924
MIT License
301 stars 75 forks source link

About image reading format when training #39

Closed DCC-lzhy closed 3 years ago

DCC-lzhy commented 3 years ago

In https://github.com/lukemelas/EfficientNet-PyTorch, its image reading package is PIL.Image, and it's RGB mode. But in your code, it's opencv-python and it's BGR mode. I think it doesn't impact the final performance, whatever if use the pretrained weights. And RGB reading mode will make training converge quickly.

lucastabelini commented 3 years ago

Yes, the image should be converted to RGB instead of BGR.