mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-03-13 04:45:32 +08:00
feat(benchmark): add multi-object retrieval benchmark with SAM segmentation
This commit is contained in:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user