Add stubs for pycocotools

pyrightconfig.stricter.json

@@ -59,6 +59,7 @@
         "stubs/peewee",
         "stubs/psutil",
         "stubs/psycopg2",
+        "stubs/pycocotools",
         "stubs/pyflakes",
         "stubs/Pygments",
         "stubs/PyMySQL",

stubs/pycocotools/METADATA.toml

@@ -0,0 +1,4 @@
+version = "2.0.*"
+
+[tool.stubtest]
+ignore_missing_stub = false

stubs/pycocotools/pycocotools/__init__.pyi

@@ -0,0 +1,3 @@
+from .coco import *
+from .cocoeval import *
+from .mask import *

stubs/pycocotools/pycocotools/coco.pyi

@@ -0,0 +1,46 @@
+from _typeshed import Incomplete, Self
+from pathlib import Path
+from typing import overload
+from typing_extensions import Literal, TypeAlias
+
+from .coco_types import _RLE, _Annotation, _AnnotationG, _Category, _Dataset, _EncodedRLE, _Image, _TPolygonSegmentation
+
+# import numpy as np
+# import numpy.typing as npt
+
+PYTHON_VERSION: Incomplete
+_NDArray: TypeAlias = Incomplete
+
+class COCO:
+    anns: dict[int, _Annotation]
+    dataset: _Dataset
+    cats: dict[int, _Category]
+    imgs: dict[int, _Image]
+    imgToAnns: dict[int, list[_Annotation]]
+    catToImgs: dict[int, list[int]]
+    def __init__(self, annotation_file: str | Path = ...) -> None: ...
+    def createIndex(self) -> None: ...
+    def info(self) -> None: ...
+    def getAnnIds(
+        self, imgIds: list[int] | int = ..., catIds: list[int] | int = ..., areaRng: list[float] = ..., iscrowd: bool | None = ...
+    ) -> list[int]: ...
+    def getCatIds(
+        self, catNms: list[str] | str = ..., supNms: list[str] | str = ..., catIds: list[int] | int = ...
+    ) -> list[int]: ...
+    def getImgIds(self, imgIds: list[int] | int = ..., catIds: list[int] | int = ...) -> list[int]: ...
+    def loadAnns(self, ids: list[int] | int = ...) -> list[_Annotation]: ...
+    def loadCats(self, ids: list[int] | int = ...) -> list[_Category]: ...
+    def loadImgs(self, ids: list[int] | int = ...) -> list[_Image]: ...
+    def showAnns(self, anns: list[_Annotation], draw_bbox: bool = ...) -> None: ...
+    def loadRes(self: Self, resFile: str) -> Self: ...
+    def download(self, tarDir: str | None = ..., imgIds: list[int] = ...) -> Literal[-1] | None: ...
+    def loadNumpyAnnotations(self, data: _NDArray) -> list[_Annotation]: ...
+    # def loadNumpyAnnotations(self, data: npt.NDArray[np.float64]) -> list[_Annotation]: ...
+    @overload
+    def annToRLE(self, ann: _AnnotationG[_RLE]) -> _RLE: ...
+    @overload
+    def annToRLE(self, ann: _AnnotationG[_EncodedRLE]) -> _EncodedRLE: ...
+    @overload
+    def annToRLE(self, ann: _AnnotationG[_TPolygonSegmentation]) -> _EncodedRLE: ...
+    def annToMask(self, ann: _Annotation) -> _NDArray: ...
+    # def annToMask(self, ann: _Annotation) -> npt.NDArray[np.uint8]: ...

stubs/pycocotools/pycocotools/coco_types.pyi

@@ -0,0 +1,71 @@
+from _typeshed import Incomplete
+from typing import Generic, TypeVar
+from typing_extensions import TypeAlias, TypedDict
+
+# import numpy as np
+# import numpy.typing as npt
+
+_NDArray: TypeAlias = Incomplete
+
+class _Image(TypedDict):
+    id: int
+    width: int
+    height: int
+    file_name: str
+
+_TPolygonSegmentation: TypeAlias = list[list[float]]
+
+class _RLE(TypedDict):
+    size: list[int]
+    counts: list[int]
+
+class _EncodedRLE(TypedDict):
+    size: list[int]
+    counts: str | bytes
+
+class _Annotation(TypedDict):
+    id: int
+    image_id: int
+    category_id: int
+    segmentation: _TPolygonSegmentation | _RLE | _EncodedRLE
+    area: float
+    bbox: list[float]
+    iscrowd: int  # Either 1 or 0
+
+_TSeg = TypeVar("_TSeg", _TPolygonSegmentation, _RLE, _EncodedRLE)
+
+class _AnnotationG(TypedDict, Generic[_TSeg]):
+    id: int
+    image_id: int
+    category_id: int
+    segmentation: _TSeg
+    area: float
+    bbox: list[float]
+    iscrowd: int  # Either 1 or 0
+
+class _Category(TypedDict):
+    id: int
+    name: str
+    supercategory: str
+
+class _EvaluationResult(TypedDict):
+    image_id: int
+    category_id: int
+    aRng: list[int]
+    maxDet: int
+    dtIds: list[int]
+    gtIds: list[int]
+    dtMatches: _NDArray
+    # dtMatches: npt.NDArray[np.float64]
+    gtMatches: _NDArray
+    # gtMatches: npt.NDArray[np.float64]
+    dtScores: list[float]
+    gtIgnore: _NDArray
+    # gtIgnore: npt.NDArray[np.float64]
+    dtIgnore: _NDArray
+    # dtIgnore: npt.NDArray[np.float64]
+
+class _Dataset(TypedDict):
+    images: list[_Image]
+    annotations: list[_Annotation]
+    categories: list[_Category]

stubs/pycocotools/pycocotools/cocoeval.pyi

@@ -0,0 +1,48 @@
+from _typeshed import Incomplete
+from typing_extensions import Literal, TypeAlias
+
+from .coco import COCO
+from .coco_types import _EvaluationResult
+
+# import numpy as np
+# import numpy.typing as npt
+
+_NDArray: TypeAlias = Incomplete
+_TIOU: TypeAlias = Literal["segm", "bbox", "keypoints"]
+
+class COCOeval:
+    cocoGt: COCO
+    cocoDt: COCO
+    evalImgs: list[_EvaluationResult]
+    eval: _EvaluationResult
+    params: Params
+    stats: _NDArray
+    # stats: npt.NDArray[np.float64]
+    ious: dict[tuple[int, int], list[float]]
+    def __init__(self, cocoGt: COCO | None = ..., cocoDt: COCO | None = ..., iouType: _TIOU = ...) -> None: ...
+    def evaluate(self) -> None: ...
+    def computeIoU(self, imgId: int, catId: int) -> list[float]: ...
+    def computeOks(self, imgId: int, catId: int) -> _NDArray: ...
+    # def computeOks(self, imgId: int, catId: int) -> npt.NDArray[np.float64]: ...
+    def evaluateImg(self, imgId: int, catId: int, aRng: list[int], maxDet: int) -> _EvaluationResult: ...
+    def accumulate(self, p: Params | None = ...) -> None: ...
+    def summarize(self) -> None: ...
+
+class Params:
+    imgIds: list[int]
+    catIds: list[int]
+    iouThrs: _NDArray
+    # iouThrs: npt.NDArray[np.float64]
+    recThrs: _NDArray
+    # recThrs: npt.NDArray[np.float64]
+    maxDets: list[int]
+    areaRng: list[int]
+    areaRngLbl: list[str]
+    useCats: int
+    kpt_oks_sigmas: _NDArray
+    # kpt_oks_sigmas: npt.NDArray[np.float64]
+    iouType: _TIOU
+    useSegm: int | None
+    def __init__(self, iouType: _TIOU = ...) -> None: ...
+    def setDetParams(self) -> None: ...
+    def setKpParams(self) -> None: ...

stubs/pycocotools/pycocotools/mask.pyi

@@ -0,0 +1,28 @@
+from _typeshed import Incomplete
+from typing import overload
+from typing_extensions import TypeAlias
+
+from .coco_types import _EncodedRLE
+
+# import numpy as np
+# import numpy.typing as npt
+
+_NPUInt32: TypeAlias = Incomplete  # np.uint32
+_NDArrayUInt8: TypeAlias = Incomplete  # npt.NDArray[np.uint8]
+_NDArrayUInt32: TypeAlias = Incomplete  # npt.NDArray[np.uint32]
+_NDArrayFloat64: TypeAlias = Incomplete  # npt.NDArray[np.float64]
+
+def iou(
+    dt: _NDArrayUInt32 | list[float] | list[_EncodedRLE],
+    gt: _NDArrayUInt32 | list[float] | list[_EncodedRLE],
+    pyiscrowd: list[int] | _NDArrayUInt8,
+) -> list | _NDArrayFloat64: ...
+def merge(rleObjs: list[_EncodedRLE], intersect: int = ...): ...
+@overload
+def frPyObjects(pyobj: _NDArrayUInt32 | list[list[int]] | list[_EncodedRLE], h: int, w: int) -> list[_EncodedRLE]: ...
+@overload
+def frPyObjects(pyobj: list[int] | _EncodedRLE, h: int, w: int) -> _EncodedRLE: ...
+def encode(bimask: _NDArrayUInt8) -> _EncodedRLE: ...
+def decode(rleObjs: _EncodedRLE) -> _NDArrayUInt8: ...
+def area(rleObjs: _EncodedRLE) -> _NPUInt32: ...
+def toBbox(rleObjs: _EncodedRLE) -> _NDArrayFloat64: ...