feat(benchmark): add multi-object retrieval benchmark with SAM segmentation

This commit is contained in:
2026-03-12 12:51:04 +08:00
parent 2466ab28cd
commit b39ee74e99
7 changed files with 1058 additions and 0 deletions

View File

@@ -118,6 +118,17 @@ class BenchmarkTaskConfig(BaseModel):
type: str = Field(default="retrieval", description="Task type")
top_k: int = Field(default=10, gt=0, description="Top K for recall evaluation")
# Multi-object retrieval specific settings
gamma: float = Field(
default=1.0, ge=0, description="Co-occurrence penalty exponent"
)
top_k_per_object: int = Field(
default=50, gt=0, description="Top K results per object query"
)
num_query_objects: int = Field(
default=3, gt=0, description="Number of objects to sample from query image"
)
class BenchmarkConfig(BaseModel):
"""Configuration for benchmark evaluation."""