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

Implemented v9-t model variant #137

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

cansik
Copy link

@cansik cansik commented Dec 11, 2024

I have implemented the tiny (v9-t) model variant of the v9 model, based on the original v9-s model description. I haven't trained it on COCO but on a custom dataset, which appeared to perform as expected.

Here are the model statistics (for single-class training):

┏━━━┳━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━┳━━━━━━━┓
┃   ┃ Name   ┃ Type                 ┃ Params ┃ Mode  ┃
┡━━━╇━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━╇━━━━━━━┩
│ 0 │ model  │ YOLO                 │  2.7 M │ train │
│ 1 │ metric │ MeanAveragePrecision │      0 │ train │
│ 2 │ ema    │ YOLO                 │  2.7 M │ train │
└───┴────────┴──────────────────────┴────────┴───────┘
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━┓
┃ Attributes                             ┃ Value      ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━┩
│ Trainable params                       │ 2.7 M      │
│ Non-trainable params                   │ 2.7 M      │
│ Total params                           │ 5.3 M      │
│ Total estimated model params size (MB) │ 21         │
│ Modules in train mode                  │ 2967       │
│ Modules in eval mode                   │ 0          │
└────────────────────────────────────────┴────────────┘

Related to #92

@zsp1987
Copy link

zsp1987 commented Feb 4, 2025

@cansik do you have a trained model?

@cansik
Copy link
Author

cansik commented Feb 5, 2025

@zsp1987 Not trained on COCO, I used it for a custom dataset.

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.

2 participants