Megvii-BaseDetection / YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Apache License 2.0
9.14k stars 2.15k forks source link

ImageNet Pretraining #1738

Open ocetintas opened 7 months ago

ocetintas commented 7 months ago

Hi, I am looking for ImageNet pretrained weights of the YOLOX backbone. I am specifically interested in the largest model YOLOX-x. In a couple of other issues I've seen that nano version can be trained from scratch for the same performance, but there is no information about the x version.

Overall, does ImageNet training not help at all? If there are no pretrained weights but ImageNet pretraining does help, I will invest time into training the model myself. So, I'd really appreciate your help @FateScript

Thank you in advance

FateScript commented 7 months ago

Hi, @ocetintas If you train your network with enough epoch/iter/data, the performance will be nearly the same. The effect of imagenet pretraining is making your network achieve a modest performance with less epoch/iter/data. However, to achieve a best performance, imagenet pretraining couldn't help much.