Skip to content

Commit aeed08e

Browse files
authored
Docs: update pycocotools url in CocoDetection and CocoCaptions. (#8932)
1 parent 111aafb commit aeed08e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torchvision/datasets/coco.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010
class CocoDetection(VisionDataset):
1111
"""`MS Coco Detection <https://cocodataset.org/#detection-2016>`_ Dataset.
1212
13-
It requires the `COCO API to be installed <https://github.com/pdollar/coco/tree/master/PythonAPI>`_.
13+
It requires `pycocotools <https://github.com/ppwwyyxx/cocoapi>`_ to be installed,
14+
which could be installed via ``pip install pycocotools`` or ``conda install conda-forge::pycocotools``.
1415
1516
Args:
1617
root (str or ``pathlib.Path``): Root directory where images are downloaded to.
@@ -65,7 +66,8 @@ def __len__(self) -> int:
6566
class CocoCaptions(CocoDetection):
6667
"""`MS Coco Captions <https://cocodataset.org/#captions-2015>`_ Dataset.
6768
68-
It requires the `COCO API to be installed <https://github.com/pdollar/coco/tree/master/PythonAPI>`_.
69+
It requires `pycocotools <https://github.com/ppwwyyxx/cocoapi>`_ to be installed,
70+
which could be installed via ``pip install pycocotools`` or ``conda install conda-forge::pycocotools``.
6971
7072
Args:
7173
root (str or ``pathlib.Path``): Root directory where images are downloaded to.

0 commit comments

Comments
 (0)