fix the rgb <> bgr issue on model training inference - Githubissues

picselliahq / picsellia-training-engine

MIT License

2 stars 0 forks source link

fix the rgb <> bgr issue on model training inference #159

Closed picsalex closed 2 weeks ago

picsalex commented 3 weeks ago

Description

The RGB<>BGR should now be resolved:

For ultralytics, the conversion is made on the evaluation step, since we are working with numpy arrays (ultralytics does an implicit BGR <> RGB conversion because they thing it comes from an OpenCV opening ith cv2.imread).
For YOLOX, the conversion is made on the training step, since they use cv2.imread everywhere.

Issues

This PR resolves #158