Skip to content

Commit 557554f

Browse files
[pycocotools] Fix COCOeval.eval and Params.areaRng (#9897)
* Rename `_EvaluationResult` to `_ImageEvaluationResult` * Add new `_EvaluationResult` * Fix type of `COCOeval.eval` * Correct Params.areaRng which is in fact a list of pairs
1 parent 3a82796 commit 557554f

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

stubs/pycocotools/pycocotools/cocoeval.pyi

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ from .coco import COCO
1010
_NDArray: TypeAlias = Incomplete
1111
_TIOU: TypeAlias = Literal["segm", "bbox", "keypoints"]
1212

13-
class _EvaluationResult(TypedDict):
13+
class _ImageEvaluationResult(TypedDict):
1414
image_id: int
1515
category_id: int
1616
aRng: list[int]
@@ -27,10 +27,21 @@ class _EvaluationResult(TypedDict):
2727
dtIgnore: _NDArray
2828
# dtIgnore: npt.NDArray[np.float64]
2929

30+
class _EvaluationResult(TypedDict):
31+
params: Params
32+
counts: list[int]
33+
date: str
34+
# precision: npt.NDArray[np.float64]
35+
precision: _NDArray
36+
# recall: npt.NDArray[np.float64]
37+
recall: _NDArray
38+
# scores: npt.NDArray[np.float64]
39+
scores: _NDArray
40+
3041
class COCOeval:
3142
cocoGt: COCO
3243
cocoDt: COCO
33-
evalImgs: list[_EvaluationResult]
44+
evalImgs: list[_ImageEvaluationResult]
3445
eval: _EvaluationResult
3546
params: Params
3647
stats: _NDArray
@@ -41,7 +52,7 @@ class COCOeval:
4152
def computeIoU(self, imgId: int, catId: int) -> list[float]: ...
4253
def computeOks(self, imgId: int, catId: int) -> _NDArray: ...
4354
# def computeOks(self, imgId: int, catId: int) -> npt.NDArray[np.float64]: ...
44-
def evaluateImg(self, imgId: int, catId: int, aRng: list[int], maxDet: int) -> _EvaluationResult: ...
55+
def evaluateImg(self, imgId: int, catId: int, aRng: list[int], maxDet: int) -> _ImageEvaluationResult: ...
4556
def accumulate(self, p: Params | None = ...) -> None: ...
4657
def summarize(self) -> None: ...
4758

@@ -53,7 +64,7 @@ class Params:
5364
recThrs: _NDArray
5465
# recThrs: npt.NDArray[np.float64]
5566
maxDets: list[int]
56-
areaRng: list[int]
67+
areaRng: list[list[float]]
5768
areaRngLbl: list[str]
5869
useCats: int
5970
kpt_oks_sigmas: _NDArray

0 commit comments

Comments
 (0)