Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The list of COCO_INSTANCE_CATEGORY_NAMES is wrong #990

Closed
hirune924 opened this issue Jun 5, 2019 · 1 comment
Closed

The list of COCO_INSTANCE_CATEGORY_NAMES is wrong #990

hirune924 opened this issue Jun 5, 2019 · 1 comment

Comments

@hirune924
Copy link
Contributor

In this page,
https://pytorch.org/docs/stable/torchvision/models.html#object-detection-instance-segmentation-and-person-keypoint-detection

The list of COCO_INSTANCE_CATEGORY_NAMES is wrong.
Category name include space is treated as two categories.

I think the correct category is
COCO_INSTANCE_CATEGORY_NAMES = [
'background', 'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus',
'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'N/A', 'stop sign',
'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
'elephant', 'bear', 'zebra', 'giraffe', 'N/A', 'backpack', 'umbrella', 'N/A', 'N/A',
'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball',
'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket',
'bottle', 'N/A', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl',
'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'N/A', 'dining table',
'N/A', 'N/A', 'toilet', 'N/A', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone',
'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'N/A', 'book',
'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush'
]
Is this right? Please check.

@hirune924 hirune924 changed the title In docs, the list of COCO_INSTANCE_CATEGORY_NAMES is wrong The list of COCO_INSTANCE_CATEGORY_NAMES is wrong Jun 5, 2019
@fmassa
Copy link
Member

fmassa commented Jun 5, 2019

Thanks for catching this!

Fixed by @hirune924 in #991

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants