diff --git a/src/polystar/target_pipeline/objects_filters/size_filter.py b/src/polystar/target_pipeline/objects_filters/size_filter.py new file mode 100644 index 0000000000000000000000000000000000000000..184a358b8996c4d5904d3a905d25bc19ffa2c98e --- /dev/null +++ b/src/polystar/target_pipeline/objects_filters/size_filter.py @@ -0,0 +1,12 @@ +from dataclasses import dataclass + +from polystar.models.roco_object import ROCOObject +from polystar.target_pipeline.objects_filters.objects_filter_abc import ObjectsFilterABC + + +@dataclass +class SmallObjectFilter(ObjectsFilterABC): + min_size: int + + def validate_single(self, obj: ROCOObject) -> bool: + return obj.box.area >= self.min_size