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.

Sign up for GitHub

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

Jump to bottom

General classifier training #19

Merged
merged 56 commits into from
Jun 24, 2024
Merged

General classifier training #19

merged 56 commits into from
Jun 24, 2024

Conversation

MinhxNguyen7
Copy link
Collaborator

@MinhxNguyen7 MinhxNguyen7 commented May 25, 2024 •
edited
Loading

Description

  • Code for loading data and training the general classifier.
  • Training is wrapped with LightningModule for easier training management and progress visualization.
  • Dataset uses existing parsing functionality from yolo_io.YoloReader.

Tasks

  • Single-batch train and back-propagation methods.
  • Dataset and dataloader for getting data.
  • Train script.
  • Verify that everything works.
  • Train the model.
  • Export the model weights.
  • Tune hyper-parameters.

Test Plan

  • Incorporate it into the image processor and run current tests to check performance.

Issues

@MinhxNguyen7 MinhxNguyen7 added the enhancement New feature or request label May 25, 2024
@MinhxNguyen7 MinhxNguyen7 self-assigned this May 25, 2024
@MinhxNguyen7
feat: cropping with min size
74c0c29
@MinhxNguyen7
feat: BBoxToCropTransformer and test (#14)
b14c536 
* feat: BBoxToCropTransformer and test

* fix: add data transformer param, change transformer test to file log

* delete commented chunk

* add class types and fix typing

* feat: make test file read actual data

* fix: gitignore

* wip color integration for yolo transformer

* fix: relative imports

* fix: more relative imports

* feat: add script to run crop pipeline

* feat: more accurate class type filtering

* feat: add padding to script

---------

Co-authored-by: MinhxNguyen7 <[email protected]>
@MinhxNguyen7
Merge branch 'main' into HEAD
6bc2178
@MinhxNguyen7
feat: YoloReader read single data point
62276ec
@MinhxNguyen7
fix: packages bs
a16cd1d
@MinhxNguyen7 MinhxNguyen7 marked this pull request as ready for review June 24, 2024 20:47
@MinhxNguyen7 MinhxNguyen7 merged commit 2636a40 into main Jun 24, 2024
@MinhxNguyen7 MinhxNguyen7 deleted the general-classifier branch June 24, 2024 20:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@MinhxNguyen7 MinhxNguyen7

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

General classifier training
1 participant
@MinhxNguyen7