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

[experimential] [detection] model training iteration with updated augmentation pipeline #1604

Open
9 tasks
Tracked by #1752
felixdittrich92 opened this issue May 22, 2024 · 0 comments
Open
9 tasks
Tracked by #1752
Assignees
Labels
benchmark This issue reports a benchmark framework: pytorch Related to PyTorch backend framework: tensorflow Related to TensorFlow backend topic: text detection Related to the task of text detection type: enhancement Improvement
Milestone

Comments

@felixdittrich92
Copy link
Contributor

felixdittrich92 commented May 22, 2024

🚀 The feature

Retrain detection models from scratch with the updated augmentation pipeline

No milestone yet we will do this step by step

  • db_mobilenet_v3_large (pt - done)
  • db_resnet50
  • db_resnet34
  • linknet_resnet18
  • linknet_resnet34
  • linknet_resnet50
  • fast_tiny (can be ported to TF @felixdittrich92 )
  • fast_small (can be ported to TF @felixdittrich92)
  • fast_base (can be ported to TF @felixdittrich92)

args:

--rotation
--save_interval
--epochs 20

benchmarks: @felixdittrich92

goal:

Model robustness and improvements on edge cases like:

#1535
#1547

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
benchmark This issue reports a benchmark framework: pytorch Related to PyTorch backend framework: tensorflow Related to TensorFlow backend topic: text detection Related to the task of text detection type: enhancement Improvement
Projects
None yet
Development

No branches or pull requests

2 participants