hustvl/YOLOP

Bug: Input is in RGB format, but augment_hsv() assumes BGR format

RahulRewale opened this issue · 0 comments

In the __getitem__() function in lib/dataset/AutoDriveDataset.py, we read an image and then convert from BGR to RGB. This image is then passed to augment_hsv() in lib/utils/augmentations.py file. Here, we convert from BGR to HSV and then HSV to BGR, but the input to the function is in RGB format.