Loss function for YOLOX

[guher6]

Hi,

Thanks for a nice repo. I'm looking to continue training a pretrained YOLOX on another dataset and was wondering if there's a defined loss function for YOLOX somewhere? I cannot find it and need it to be able to continue training the model on a custom dataset.

Thanks

[leondgarse]

• AnchorFreeLoss is for typical YOLOX model. You can find some basic tests there.
• Training usage in this repo can follow https://github.com/leondgarse/keras_cv_attention_models#coco-training-and-evaluating, like custom_dataset and other parameters. Ya, it's lacking detail control during training. You can also follow coco_train_script.py creating your own dataset / loss / compile / fit process.

[guher6]

Thank you! Sounds great. Will try it out.