Megvii-BaseDetection / Megvii-BaseDetection/YOLOX
Looking for help with transfer learning/fine tuning YOLOX
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 10.6k
- Forks
- 2.5k
- PR merge metrics
- No merged PRs in 30d
Description
Hello,
I would like to enhance the standard COCO trained YOLOX_tiny model with some additional images from the environment I intend to deploy it in, to improve accuracy and reduce false positives. I have seen tutorials on doing transfer learning/fine tuning on other YOLO variants like Yolov5, where the backbone is frozen, training is done, then backbone is unfrozen and some more training is done with smaller learning rates (fine tuning).
I see that the ability to freeze layers is present in YOLOX, but I haven't been able to find a write up/tutorial on which layers to freeze, what kind of learning parameters to use etc. Any help would be appreciated!
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Review YOLOX's existing layer-freezing support and the YOLOX_tiny COCO training setup first. Document which layers and learning parameters are appropriate for transfer learning and fine-tuning, including the freeze/unfreeze sequence, so users can follow a reproducible workflow.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Documentation
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100