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

Dev/update train use dataset #35

Merged
merged 20 commits into from
Oct 25, 2024

Conversation

Purg
Copy link
Member

@Purg Purg commented Oct 24, 2024

Overhauls the dataset usage in TCN-HPL to make use of truth, detection and pose input COCO files. This still maintains training functionalities, achieving the same or better accuracy on M2 at least than previous training runs, and supports much more direct utilization by down-stream users, i.e. the targetted ROS streaming inference use-case.

Includes offline and online data loading methods for different use cases
to invoke as appropriate.
Frame identifier returned has no bearing on training, but is utilized in
plotting and within the training framework. This more benefits from a
label that is reflective of frame number than the possibly arbitrary IDs
assigned to image objects in the COCO file.
Also adds a function to recreate a dataset from a given hyperparameter
config as from a training run that produced a given model. This is to be
used by the prediction code to generate a dataset such that congruent
vectorization will occur as done during training.

Updates the latest M2 experiment file as a test-bed.
@Purg Purg mentioned this pull request Oct 24, 2024
@Purg Purg merged commit 0a800a0 into PTG-Kitware:main Oct 25, 2024
0 of 11 checks passed
@Purg Purg deleted the dev/update-train-use-dataset branch October 25, 2024 00:53
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