YOLOX icon indicating copy to clipboard operation
YOLOX copied to clipboard

How to train YOLOX with RGB images

Open nahidalam opened this issue 3 years ago • 1 comments

Since YOLOX is trained on BGR images (cv2.imread returns a BGR by default), what would it take if I want to train it on RGB images?

I am thinking below steps:

  • Since my dataset is COCO format, change the dataset read here
  • Replace BGR with RGB in augment_hsv function here and here

Anything else? @FateScript

nahidalam avatar Aug 25 '22 10:08 nahidalam

@nahidalam For a long time training, RGB and BGR doesn't matter too much. I think fix bug in augment_hsv could help, but fixing this might not make the result better.

FateScript avatar Aug 29 '22 04:08 FateScript