Skip to content
Snippets Groups Projects
Commit 98387ddc authored by Mathieu Beligon's avatar Mathieu Beligon
Browse files

[filter] (size) Add SizeObjectsFilter

parent 49635732
No related branches found
No related tags found
No related merge requests found
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment