280 Commits

Author	SHA1	Message	Date
mbsantiago	716b3a3778	Allow enabling/disabling filtering and transforms during data loading	2026-03-28 20:27:04 +00:00
mbsantiago	548cd366cd	Add source filtering to dataset loading	2026-03-28 19:43:13 +00:00
mbsantiago	b253a54cc8	Use detection threshold use to API and CLI	2026-03-28 14:26:10 +00:00
mbsantiago	dfe19c68f0	Add detection threshold override option to postprocessor	2026-03-28 14:25:18 +00:00
mbsantiago	e4bbde9995	Better cli docs	2026-03-28 13:25:15 +00:00
mbsantiago	7277151f33	Start cli docs	2026-03-28 11:28:52 +00:00
mbsantiago	9ec2f1a107	Add audio dir to data cli command	2026-03-28 11:09:47 +00:00
mbsantiago	5a14b29281	Do not crash on failure to plot	2026-03-19 01:31:21 +00:00
mbsantiago	13a31d9de9	Do not sync with model in loading from checkpoint otherwise device clash	2026-03-19 01:28:05 +00:00
Santiago Martinez Balvanera	b8acd86c71	By default only save the last checkpoint	2026-03-19 01:26:11 +00:00
Santiago Martinez Balvanera	875751d340	Add save last config to checkpoint	2026-03-19 00:36:45 +00:00
Santiago Martinez Balvanera	32d8c4a9e5	Create separate preproc/postproc/target instances for model in api	2026-03-19 00:23:55 +00:00
mbsantiago	fb3dc3eaf0	Ensure preprocessor is in CPU	2026-03-19 00:09:30 +00:00
mbsantiago	0d90cb5cc3	Create duplicate preprocessor for data input pipeline	2026-03-19 00:03:29 +00:00
mbsantiago	91806aa01e	Update default scheduler params	2026-03-18 23:54:16 +00:00
mbsantiago	23ac619c50	Add option to override targets when loading the model config	2026-03-18 23:36:24 +00:00
mbsantiago	99b9e55c0e	Add extra and strict arguments to load config functions, migrate example config	2026-03-18 22:06:45 +00:00
mbsantiago	652670b01d	Name changes	2026-03-18 20:48:35 +00:00
mbsantiago	0163a572cb	Expanded cli tests	2026-03-18 20:35:08 +00:00
mbsantiago	f0af5dd79e	Change inference command to predict	2026-03-18 20:07:53 +00:00
mbsantiago	2f03abe8f6	Remove stale load functions	2026-03-18 19:43:54 +00:00
mbsantiago	22a3d18d45	Move the logging config out of the train/eval configs	2026-03-18 19:32:19 +00:00
mbsantiago	bf5b88016a	Polish evaluate and train CLI	2026-03-18 19:15:57 +00:00
mbsantiago	f9056eb19a	Ensure config is source of truth	2026-03-18 18:33:51 +00:00
mbsantiago	ebe7e134e9	Allow building new head	2026-03-18 17:44:35 +00:00
mbsantiago	8e35956007	Create save evaluation results	2026-03-18 16:49:22 +00:00
mbsantiago	a332c5c3bd	Allow loading predictions in different formats	2026-03-18 16:17:50 +00:00
mbsantiago	9fa703b34b	Allow training on existing model	2026-03-18 13:58:52 +00:00
mbsantiago	0bf809e376	Exported types at module level	2026-03-18 13:08:28 +00:00
mbsantiago	7b1cb402b4	Add a few clip and detection transforms for outputs	2026-03-18 11:24:22 +00:00
mbsantiago	45ae15eed5	Cleanup postprocessing	2026-03-18 09:07:13 +00:00
mbsantiago	b3af70761e	Create EvaluateTaskProtocol	2026-03-18 01:53:28 +00:00
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