Skip to content

Commit

Permalink
fix anchor values dtype
Browse files Browse the repository at this point in the history
  • Loading branch information
jkbmrz committed Mar 15, 2024
1 parent c63b752 commit ecee414
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ class HeadObjectDetection(Head, ABC):
description="Confidence score threshold above which a detected object is considered valid."
)
max_det: int = Field(description="Maximum detections per image.")
anchors: Optional[List[List[List[int]]]] = Field(
anchors: Optional[List[List[List[float]]]] = Field(
None,
description="Predefined bounding boxes of different sizes and aspect ratios. The innermost lists are length 2 tuples of box sizes. The middle lists are anchors for each output. The outmost lists go from smallest to largest output.",
)
Expand Down

0 comments on commit ecee414

Please sign in to comment.