Megvii-BaseDetection / Megvii-BaseDetection/YOLOX

Looking for help with transfer learning/fine tuning YOLOX

Open
#1,604 2 comments 0 reactions 0 assignees View on GitHub

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

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.