[MAJOR] Improve dataset preprocessing, upgrade onnx and tensorrt api, bugfixes - #40

Merged

PeterCalifano merged 38 commits into

mainfrom

feature/extend_onnx_trt_api

Mar 4, 2026

Commits on Feb 7, 2026

Update and cleanup implementation of base models
PeterCalifano
committed
688b21a
Add placeholder implementation of augs manager (WIP)
PeterCalifano
committed
be6858a
Add initial implementation of 1D augs model classes
PeterCalifano
committed
577e072
Merge branch 'main' into feature/upgrade_augs_module
PeterCalifano
committed
dd1e3d6
Merge branch 'dev_main' into feature/upgrade_augs_module
PeterCalifano
committed
7b6bb60
Update environment setup and caching
PeterCalifano
committed
09cb355
Update TODO file
PeterCalifano
committed
25fa041
Merge branch 'main' into dev_main
PeterCalifano
committed
ad0a282
Merge branch 'dev_main' into feature/upgrade_augs_module
PeterCalifano
committed
f14a8b1

Commits on Feb 14, 2026

Implement fetching of affine transform from geom augs
PeterCalifano
committed
8e2ab39
Implement static method for 1D angle label update
PeterCalifano
committed
a6a5664
Complete extension of tests for external usage
PeterCalifano
committed
2f3d70d
Improve resize adapter
PeterCalifano
committed
6f5a0fa
Improve configuration options
PeterCalifano
committed
4a8ed47
Change default to false
PeterCalifano
committed
d4304fc
[BUGFIX] Rework paths fetching function
PeterCalifano
committed
223eb87

Commits on Feb 16, 2026

Update pyTorchAutoForge/datasets/vector_error_models/VectorErrorsBaseClasses.py

PeterCalifano
and
Copilot
authored
33293eb
Update pyTorchAutoForge/datasets/noise_models/FrameCameraErrorModels.py

PeterCalifano
and
Copilot
authored
080ecff
Update pyTorchAutoForge/datasets/AugmentationsManager.py

PeterCalifano
and
Copilot
authored
5629a27
Update pyTorchAutoForge/datasets/AugmentationsBaseClasses.py

PeterCalifano
and
Copilot
authored
2f6c965
Update tests/datasets/test_VectorErrorsBaseClasses.py

PeterCalifano
and
Copilot
authored
e847d8d
Merge pull request #37 from PeterCalifano/feature/upgrade_augs_module
PeterCalifano
authored
47cc64f
Update torch requirement
PeterCalifano
committed
9ef8eae

Commits on Feb 18, 2026

Add AGENTS.md
PeterCalifano
committed
464067d
Extend dataset base class image loading
PeterCalifano
committed
c4c35d7
Merge pull request #38 from PeterCalifano/feature/extend_dataset_interface_rgb_grayscale
PeterCalifano
authored
16cf71b

Commits on Feb 21, 2026

Rework FetchDatasetPaths for efficiency
PeterCalifano
committed
d6efc3a
Switch from none prealloc to append (faster)
PeterCalifano
committed
d5c3441
Implement process-based parallelization for preprocessing
PeterCalifano
committed
e5ce358
Wire new option to dataset classes
PeterCalifano
committed
7e23a22
Merge pull request #39 from PeterCalifano/optimization/datasets_prepro
PeterCalifano
authored
6e4f188
[HOTIX] Rework dataset fetching preprocessing
PeterCalifano
committed
1311506

Commits on Mar 2, 2026

Add tuple functionality
Umberto
committed
f540ce2

Commits on Mar 3, 2026

Commits on Mar 4, 2026

Update tests/datasets/test_VectorErrorsBaseClasses.py

PeterCalifano
and
Copilot
authored
172bbb8