Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add YOLOX support #2644

Merged
merged 299 commits into from
Jan 13, 2023
Merged

Conversation

FANGAreNotGnu
Copy link
Contributor

@FANGAreNotGnu FANGAreNotGnu commented Jan 5, 2023

  • Add support for YOLOX (yolox-s and yolox-l)
  • Update finetuning example to include yolox
  • Update code loading customized detection configs: store locally in multimodal/configs/pretrain/detection instead of download from s3.
  • Add max_tuning_num in fit() to control the max number of data to use for validation.
  • Add clean_old_ckpts flag to control if we shall clean the old ckpts.

YoloX-L performance on pothole dataset:
training time: 1337 seconds
mAP: 0.449
mAP50: 0.736

Previous best (VFNet-R50):
training time: ~18000 seconds
mAP: 0.450
mAP50: 0.738

TODO:

  • Test the YOLOX pretrained on objects365
  • More pretrain tunings.
  • Test YOLOX-tiny/s/m/x
  • update tutorial to use YOLOX

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@FANGAreNotGnu FANGAreNotGnu changed the title [WIP] Add YOLOX support Add YOLOX support Jan 12, 2023
@github-actions
Copy link

Job PR-2644-08a8c29 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-2644/08a8c29/index.html

@github-actions
Copy link

Job PR-2644-6ec55ff is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-2644/6ec55ff/index.html

@github-actions
Copy link

Job PR-2644-5079fd5 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-2644/5079fd5/index.html

Copy link
Contributor

@bryanyzhu bryanyzhu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks for bringing yolox in.

Copy link
Contributor

@zhiqiangdon zhiqiangdon left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@FANGAreNotGnu FANGAreNotGnu merged commit cc50c7d into autogluon:master Jan 13, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants