Commit Graph

248 Commits

Author SHA1 Message Date
mbsantiago
daff74fdde Moved decoding to outputs 2026-03-18 01:35:34 +00:00
mbsantiago
31d4f92359 reformat 2026-03-18 00:41:45 +00:00
mbsantiago
d56b9f02ae Remove num_worker from config 2026-03-18 00:22:59 +00:00
mbsantiago
573d8e38d6 Ran formatter 2026-03-18 00:03:26 +00:00
mbsantiago
751be53edf Moving types around to each submodule 2026-03-18 00:01:35 +00:00
mbsantiago
c226dc3f2b Add outputs module 2026-03-17 23:00:26 +00:00
mbsantiago
3b47c688dd Update api_v2 2026-03-17 22:25:41 +00:00
mbsantiago
feee2bdfa3 Add scheduler and optimizer module 2026-03-17 21:16:41 +00:00
mbsantiago
615c7d78fb Added a full test of training and saving 2026-03-17 15:38:07 +00:00
mbsantiago
56f6affc72 use train config in training module 2026-03-17 14:16:40 +00:00
mbsantiago
65bd0dc6ae Restructure model config 2026-03-17 13:33:13 +00:00
mbsantiago
1a7c0b4b3a Removing legacy types 2026-03-17 12:53:03 +00:00
mbsantiago
8ac4f4c44d Add dynamic imports to existing registries 2026-03-16 10:04:34 +00:00
mbsantiago
038d58ed99 Add config for dynamic imports, and tests 2026-03-16 09:30:23 +00:00
mbsantiago
e0503487ec Remove deprecated types 2026-03-15 21:17:20 +00:00
mbsantiago
4b7d23abde Create data annotation loader registry 2026-03-15 20:53:59 +00:00
mbsantiago
0d590a26cc Update tests 2026-03-08 18:41:05 +00:00
mbsantiago
46c02962f3 Add test for preprocessing 2026-03-08 17:11:27 +00:00
mbsantiago
bfc88a4a0f Update audio loader docstrings 2026-03-08 17:02:35 +00:00
mbsantiago
ce952e364b Update audio module docstrings 2026-03-08 17:02:25 +00:00
mbsantiago
ef3348d651 Update model docstrings 2026-03-08 16:34:17 +00:00
mbsantiago
4207661da4 Add test for backbones 2026-03-08 16:04:54 +00:00
mbsantiago
45e3cf1434 Modify example config to add name 2026-03-08 15:23:14 +00:00
mbsantiago
f2d5088bec Run formatter 2026-03-08 15:18:21 +00:00
mbsantiago
652d076b46 Add backbone registry 2026-03-08 15:02:56 +00:00
mbsantiago
e393709258 Add interfaces for encoder/decoder/bottleneck 2026-03-08 14:43:16 +00:00
mbsantiago
54605ef269 Add blocks and detector tests 2026-03-08 14:17:47 +00:00
mbsantiago
6812e1c515 Update default detection class config 2026-03-08 13:17:23 +00:00
mbsantiago
d52e988b8f Fix type errors 2026-03-08 12:55:36 +00:00
mbsantiago
cce1b49a8d Run formatting 2026-03-08 08:59:28 +00:00
mbsantiago
8313fe1484 Minor formatting 2026-03-07 16:09:43 +00:00
mbsantiago
4509602e70 Run automated fixes 2025-12-12 21:29:25 +00:00
mbsantiago
0adb58e039 Run formatter 2025-12-12 21:28:28 +00:00
mbsantiago
531ff69974 Cleanup evaluate 2025-12-12 21:14:31 +00:00
mbsantiago
750f9e43c4 Make sure threshold is used 2025-12-12 19:53:15 +00:00
mbsantiago
f71fe0c2e2 Using matching and affinity functions from soundevent 2025-12-12 19:25:01 +00:00
mbsantiago
113f438e74 Run lint fixes 2025-12-08 17:19:33 +00:00
mbsantiago
2563f26ed3 Update type hints to python 3.10 2025-12-08 17:14:50 +00:00
mbsantiago
9c72537ddd Add parquet format for outputs 2025-12-08 17:11:35 +00:00
mbsantiago
202c6cbab0 Fix tests 2025-11-28 18:10:37 +00:00
mbsantiago
7336638fa9 Add average precision computation to pr curves if needed 2025-11-22 00:34:42 +00:00
mbsantiago
16c401b1da Add method to compute attention weights 2025-11-22 00:34:31 +00:00
mbsantiago
4ecbc2b734 Add function to facilitate task evaluation 2025-11-22 00:34:23 +00:00
mbsantiago
dbd2d30ead save predictions as individual nc files for speed 2025-11-19 13:54:57 +00:00
mbsantiago
bdb9e18964 Add explicit kwarg name to _freq_to_index to avoid confusion 2025-11-16 23:57:11 +00:00
mbsantiago
a4498cfd83 Add functional versions of metric and plotting utils 2025-11-16 21:37:47 +00:00
mbsantiago
960b9a92e4 Fix legacy import to use reproducible UUIDs 2025-11-16 21:37:33 +00:00
mbsantiago
76503fbd12 Fix order of config merge 2025-11-12 00:17:25 +00:00
mbsantiago
69921f258a Add GreedyAffinityMatching as an alternative to optimal affinity matching 2025-11-10 19:28:47 +00:00
mbsantiago
6039b2c3eb Avoid empty strings as tags in legacy format conversion 2025-11-10 19:28:15 +00:00
mbsantiago
efc996a0db Add targets to data summary command 2025-11-10 19:27:31 +00:00
mbsantiago
110432bd40 Added batch_size and num_workers to API 2025-10-29 12:13:16 +00:00
mbsantiago
401a3832ce Add other output formats 2025-10-27 19:39:20 +00:00
mbsantiago
5736421023 Add arguments to train cli 2025-10-14 18:19:33 +01:00
mbsantiago
3913d2d350 Add merge config option 2025-10-14 18:05:12 +01:00
mbsantiago
24dcc47e73 Fix import of DEFAULT_CHECKPOINT_DIR 2025-10-04 15:16:11 +01:00
mbsantiago
160cb6ae30 Add checkpoint config 2025-10-04 15:08:34 +01:00
mbsantiago
2d796394f6 Store anns and preds instead of evals in EvaluatorModule 2025-09-30 19:10:07 +01:00
mbsantiago
49ec1916ce Fix API after module name change 2025-09-30 18:24:06 +01:00
mbsantiago
8727edf466 Save checkpoint on max class mAP 2025-09-30 18:20:31 +01:00
mbsantiago
2f48c58de1 api_v2 2025-09-30 13:56:25 +01:00
mbsantiago
981e37c346 Writing batch inference code 2025-09-30 13:22:03 +01:00
mbsantiago
30159d64a9 Update example config 2025-09-28 16:22:21 +01:00
mbsantiago
c9f0c5c431 Added bbox iou affinity function 2025-09-28 16:08:21 +01:00
mbsantiago
10865ee600 Re-org gallery example plots 2025-09-28 15:45:48 +01:00
mbsantiago
87ed44c8f7 Plotting reorganised 2025-09-27 23:58:06 +01:00
mbsantiago
df2abff654 Task/Metrics restructure 2025-09-26 15:23:38 +01:00
mbsantiago
d6ddc4514c Better evaluation organisation 2025-09-25 17:48:29 +01:00
mbsantiago
4cd983a2c2 Better train cli arg names 2025-09-18 09:44:27 +01:00
mbsantiago
e65df81db2 Evaluate using Lightning too to handle device changes 2025-09-18 09:28:21 +01:00
mbsantiago
6c25787123 Logging is not just for training 2025-09-18 09:27:40 +01:00
mbsantiago
8c80402f08 Move clips and audio to dedicated module 2025-09-18 09:27:24 +01:00
mbsantiago
b81a882b58 Add metrics and plots 2025-09-17 10:30:24 +01:00
mbsantiago
957c0735d2 Starting new API 2025-09-16 19:39:30 +01:00
mbsantiago
bbb96b33a2 Config restructuring 2025-09-16 18:57:56 +01:00
mbsantiago
7d6cba5465 Restructuring 2025-09-16 13:38:38 +01:00
mbsantiago
60e922d565 Use pascal voc map computation by default 2025-09-16 10:56:37 +01:00
mbsantiago
704b28292b Cleaning train module 2025-09-15 16:50:08 +01:00
mbsantiago
e752e96b93 Restructure eval metrics and plotting 2025-09-15 16:01:15 +01:00
mbsantiago
ec1c0ff020 Better matching module, remove generic from classification evaluations 2025-09-14 18:16:59 +01:00
mbsantiago
8628133fd7 Compute mAP 2025-09-14 10:08:51 +01:00
mbsantiago
d80377981e Fix plotting 2025-09-14 09:38:45 +01:00
mbsantiago
ad5293e0d0 Ad FileNotFoundError to plotting 2025-09-13 20:05:42 +01:00
mbsantiago
01e7a5df25 Add ignore at ends when evaluating 2025-09-13 19:03:40 +01:00
mbsantiago
6d70140bc9 Default to normal anchor 2025-09-13 13:56:47 +01:00
mbsantiago
4fd2e84773 Fix clip missalignment in validation dataset 2025-09-11 11:09:59 +01:00
mbsantiago
e65d5a6846 Added more clipping options for validation 2025-09-10 21:09:51 +01:00
mbsantiago
615c811bb4 Add detection_class_name to targets protocol 2025-09-09 20:30:20 +01:00
mbsantiago
41b18c3f0a Better order for checkpoints 2025-09-09 15:56:46 +01:00
mbsantiago
16a0fa7b75 Add targets to train cli 2025-09-09 15:45:00 +01:00
mbsantiago
115084fd2b Updat lightning version 2025-09-09 15:31:40 +01:00
mbsantiago
951dc59718 Add seed option to train 2025-09-09 13:23:56 +01:00
mbsantiago
3376be06a4 Add experiment name 2025-09-09 09:02:25 +01:00
mbsantiago
cd4955d4f3 Eval 2025-09-08 22:04:30 +01:00
mbsantiago
c73984b213 Small fixes 2025-09-08 18:35:02 +01:00
mbsantiago
d8d2e5a2c2 Remove preprocessing modules 2025-09-08 18:11:58 +01:00
mbsantiago
b056d7d28d Make sure training is still working 2025-09-08 18:03:56 +01:00
mbsantiago
95a884ea16 Update tests 2025-09-08 18:00:17 +01:00
mbsantiago
b7ae526071 Big changes in data module 2025-09-08 17:50:25 +01:00
mbsantiago
cf6d0d1ccc Remove stale tests 2025-09-07 11:03:46 +01:00