mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-01-10 17:19:34 +01:00
30 lines
725 B
Python
30 lines
725 B
Python
"""Main entry point for the BatDetect2 Postprocessing pipeline."""
|
|
|
|
from batdetect2.postprocess.config import (
|
|
PostprocessConfig,
|
|
load_postprocess_config,
|
|
)
|
|
from batdetect2.postprocess.decoding import (
|
|
convert_raw_predictions_to_clip_prediction,
|
|
to_raw_predictions,
|
|
)
|
|
from batdetect2.postprocess.nms import (
|
|
non_max_suppression,
|
|
)
|
|
from batdetect2.postprocess.postprocessor import (
|
|
Postprocessor,
|
|
build_postprocessor,
|
|
get_raw_predictions,
|
|
)
|
|
|
|
__all__ = [
|
|
"PostprocessConfig",
|
|
"Postprocessor",
|
|
"build_postprocessor",
|
|
"convert_raw_predictions_to_clip_prediction",
|
|
"to_raw_predictions",
|
|
"load_postprocess_config",
|
|
"non_max_suppression",
|
|
"get_raw_predictions",
|
|
]
|