248 Commits

Author	SHA1	Message	Date
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
mbsantiago	709b6355c2	torch.multiprocessing didn't work, returning to serial processing	2025-09-01 11:27:23 +01:00
mbsantiago	db2ad11743	Make matching in parallel for speedup	2025-09-01 11:19:02 +01:00
mbsantiago	e0ecc3c3d1	Clear evaluation callback after epoch ends	2025-09-01 08:56:38 +01:00
mbsantiago	71c2301c21	Independent preprocessor for generating validation plots	2025-08-31 23:04:16 +01:00
mbsantiago	d3d2a28130	Move detections array to cpu	2025-08-31 22:59:06 +01:00
mbsantiago	5b9a5a968f	Refactor eval code	2025-08-31 22:57:02 +01:00
mbsantiago	8d093c3ca2	Use custom AugmentationSequence instead of nn.Sequential	2025-08-31 19:27:15 +01:00
mbsantiago	2f4edeffff	Create separate preprocessor for the train/val datasets	2025-08-31 19:22:45 +01:00
mbsantiago	cca1d82d63	Remove blocking .to(device)	2025-08-31 19:10:06 +01:00
mbsantiago	55f473c9ca	Fix logging issue	2025-08-31 19:06:09 +01:00
mbsantiago	40f6b64611	Remove train preprocessing	2025-08-31 18:28:52 +01:00
mbsantiago	1cec332dd5	Change default train duration to 0.256 instead of 0.512	2025-08-30 14:08:00 +01:00
mbsantiago	93e89ecc46	LR Scheduler takes num of total batches	2025-08-28 08:52:11 +01:00
mbsantiago	34ef9e92a1	Make sure preprocessing is batchable	2025-08-27 23:58:38 +01:00
mbsantiago	0b5ac96fe8	Update model config	2025-08-27 23:58:07 +01:00
mbsantiago	ff754a1269	Tweaks of augmentation config	2025-08-27 18:23:38 +01:00
mbsantiago	ed76ec24b6	Plot anchor points	2025-08-27 18:13:40 +01:00
mbsantiago	d25efdad10	Fix plotting after update	2025-08-26 11:48:06 +01:00
mbsantiago	3043230d4f	Device fixing #5	2025-08-25 23:20:30 +01:00
mbsantiago	67e37227f5	Device fixing #4	2025-08-25 23:11:33 +01:00
mbsantiago	9d4a9fc35c	Device fixing #3	2025-08-25 23:08:49 +01:00
mbsantiago	d0bab60bf3	Device fixing #2	2025-08-25 23:07:09 +01:00
mbsantiago	a267db290c	Device fixing	2025-08-25 23:04:13 +01:00
mbsantiago	441ccb3382	Remove user warning from plotting function	2025-08-25 22:49:48 +01:00
mbsantiago	281c4dcb8a	Remove xr from postprocess	2025-08-25 22:46:21 +01:00
mbsantiago	cc9e47b022	Fix plotting after changes	2025-08-25 19:07:12 +01:00
mbsantiago	1f26103f42	Cleanup train preprocessing	2025-08-25 18:37:46 +01:00
mbsantiago	c80078feee	Removing stale tests	2025-08-25 17:23:27 +01:00
mbsantiago	0bb0caddea	Update augmentations	2025-08-25 17:06:17 +01:00
mbsantiago	76dda0a0e9	Fix train preprocessing	2025-08-25 14:01:31 +01:00
mbsantiago	c36ef3ecb5	Labels to torch	2025-08-25 12:43:34 +01:00
mbsantiago	667b18a54d	Preprocessing in pytorch	2025-08-25 11:41:55 +01:00
mbsantiago	61115d562c	Moved types to dedicated module	2025-08-24 10:55:48 +01:00
mbsantiago	02adc19070	Better structure for training module	2025-08-23 18:23:45 +01:00
mbsantiago	d4f249366e	Clean up the detector head	2025-08-18 16:07:51 +01:00
mbsantiago	98f83e8b34	Small fix	2025-08-18 10:25:14 +01:00
mbsantiago	9d1497b3f4	Handle case of empty preds or gt	2025-08-17 23:00:40 +01:00
mbsantiago	7af72912da	Implement previous matching strategy (greedy)	2025-08-17 22:52:00 +01:00
mbsantiago	4aea3fb2b0	Copy librosa PCEN implementation	2025-08-17 18:02:40 +01:00
mbsantiago	c1945ebdb7	Minor difference in loss fixed	2025-08-16 18:49:30 +01:00
mbsantiago	13da5d2814	Remove unnecessary extra conv layer	2025-08-16 18:32:35 +01:00
mbsantiago	78a0975864	Same issue	2025-08-14 11:07:27 +01:00
mbsantiago	c9848deebf	Fix logging issue	2025-08-14 11:05:21 +01:00
mbsantiago	a7301bcdc8	Add image logging for mlflow	2025-08-14 10:50:00 +01:00
mbsantiago	fdbb9c2b43	Go back to serial matching	2025-08-12 19:57:49 +01:00
mbsantiago	75e555ff00	Fix matching error after optimising	2025-08-12 19:49:24 +01:00
mbsantiago	c7b110feeb	Limit number of parallel processes to match predictions and annotations	2025-08-12 19:06:44 +01:00
mbsantiago	7d92ec772b	Fix number formatting in gallery plot	2025-08-12 18:49:21 +01:00
mbsantiago	0bb703f3c1	Add call type alias	2025-08-12 18:45:54 +01:00
mbsantiago	51d0a49da9	Improve performance of postprocessing code	2025-08-12 17:47:17 +01:00
mbsantiago	b997a122f1	Enable batdetect2 in the cli	2025-08-11 12:25:04 +01:00
mbsantiago	c41551b59c	Use match config for training	2025-08-11 12:02:46 +01:00
mbsantiago	8aa2d0cd11	Add evaluation config with match strategy	2025-08-11 12:02:30 +01:00
mbsantiago	65d13a32b7	Also ignore assertion errors in gallery plotting	2025-08-11 09:46:36 +01:00
mbsantiago	31a0d1bbb5	Drop duplicate bin edges	2025-08-11 08:59:12 +01:00
mbsantiago	f361420e58	Fix issue in gallery plotting	2025-08-11 08:54:37 +01:00
mbsantiago	2cc0bd59d4	Add plot limits	2025-08-11 01:37:22 +01:00
mbsantiago	374c62d7ab	Add dataset summary and split functions	2025-08-11 01:35:09 +01:00
mbsantiago	ef279bee5d	Update to soundevent 2.7	2025-08-11 01:34:54 +01:00
mbsantiago	2341f822a7	Ignore plotting failures in gallery plot	2025-08-08 13:30:44 +01:00
mbsantiago	d9395d3eeb	Updated callback to include plotting	2025-08-08 13:06:28 +01:00
mbsantiago	aaec66c15e	Added BatDetect2Prediction wrapper and method	2025-08-08 13:06:10 +01:00
mbsantiago	6213238585	Added matching configs	2025-08-08 13:05:50 +01:00
mbsantiago	a485ea4f79	Add get_default_preprocessor function	2025-08-08 13:03:53 +01:00
mbsantiago	3cfceb76b4	Added plotting example gallery function	2025-08-08 13:03:42 +01:00
mbsantiago	d877d383a4	Disable logs by default	2025-08-08 12:25:57 +01:00
mbsantiago	bb4a9fe645	Move legacy plot module to legacy folder	2025-08-08 12:25:49 +01:00
mbsantiago	87ce2acd6f	Adding plotting functions	2025-08-08 12:25:26 +01:00
mbsantiago	e1908c35ca	Update compat module to use new term module	2025-08-08 12:25:16 +01:00
mbsantiago	d9323a1383	Move legacy evaluate code to legacy folder	2025-08-07 16:20:16 +01:00
mbsantiago	9816985bb1	Create dataloader config	2025-07-29 12:10:51 +01:00
mbsantiago	62f0c5c397	Remove stale tensorboard logger argument	2025-07-29 11:47:07 +01:00
mbsantiago	839a632aa2	Remove non-existing option in tensorboard logger	2025-07-29 00:17:46 +01:00
mbsantiago	829c07fb12	Remove unnecessary note	2025-07-29 00:03:20 +01:00
mbsantiago	6732394f50	Add model checkpoint callback	2025-06-28 15:46:50 -06:00
mbsantiago	8368aad178	Add better logging	2025-06-28 14:45:18 -06:00
mbsantiago	19e873dd0b	Improve augmentations config and logging	2025-06-28 11:36:18 -06:00
mbsantiago	bafb9a3622	Add mlflow logger	2025-06-28 11:08:19 -06:00
mbsantiago	e8db1d4050	Fix hyperparameter saving	2025-06-26 19:53:19 -06:00
mbsantiago	b396d4908a	Fix trainer init	2025-06-26 19:40:27 -06:00
mbsantiago	ce15a0f152	Fix trainer init	2025-06-26 19:35:23 -06:00
mbsantiago	16febed792	Add test that ensures example_config is valid	2025-06-26 19:21:46 -06:00
mbsantiago	15de168a20	Update build trainer	2025-06-26 17:43:56 -06:00
mbsantiago	0c8fae4a72	Instantiate lightnign module from config	2025-06-26 17:39:50 -06:00
mbsantiago	a0622aa9a4	Fix train issues	2025-06-26 16:16:29 -06:00
mbsantiago	587742b41e	Change train to use full config	2025-06-26 16:02:41 -06:00
mbsantiago	22f7d46f46	Improve logging of train preprocessing	2025-06-26 13:08:44 -06:00
mbsantiago	1384c549f7	Create TrainPreprocessConfig	2025-06-26 12:30:16 -06:00
mbsantiago	4b6acd5e6e	Add manual logging of hyperparams	2025-06-26 11:59:33 -06:00
mbsantiago	166dad20bd	Rename BlockGroupConfig to LayerGroupConfig	2025-06-26 10:04:42 -06:00
mbsantiago	136949c4e7	Add logging config	2025-06-26 07:55:24 -06:00
mbsantiago	a462beaeb8	Remove rogue print	2025-06-24 06:27:41 -06:00
mbsantiago	daab8ff0d7	Fix validation is_in_subclip after encoder changes	2025-06-24 06:26:08 -06:00
mbsantiago	235f0e27da	Add load dataset config function	2025-06-24 12:40:29 +01:00
mbsantiago	8253b5bdc4	Update makefile	2025-06-23 18:53:15 +01:00
mbsantiago	c7ea361cf4	Implement changes needed to make roi encode/decode class dependent	2025-06-23 18:52:36 +01:00
mbsantiago	3407e1b5f0	Add other roi tests	2025-06-21 23:51:07 +01:00
mbsantiago	960558be8b	move to src layout	2025-06-21 13:48:40 +01:00