feat: 17 rework yolox data augmentation

picselliahq / picsellia-training-engine

MIT License

2 stars 0 forks source link

feat: 17 rework yolox data augmentation #151

Closed picsalex closed 3 months ago

picsalex commented 3 months ago

Description

This PR brings several new features:

An enhancement of the original Yolox model regarding data augmentation
A bug fix when YoloX, sometimes, does not log anything during the training

Issues

This PR resolves issue 17

picsalex commented 3 months ago

@SoniaGrh the file data_augment_v2 is kept for further references (it has been a part of the development process, but I decided to keep it just in case). Also, the additional requirements points to a specific commit of a fork of imgaug since the latter is deprecated with the most recent versions of numpy.

Also, the yolox-detection/experiment/test_augmentations.py file is a util script to visually test the result of the data augmentations. It's also kept for further references.