https://github.com/benediktalkin/imagenetutils
https://github.com/benediktalkin/imagenetutils
computervision imagenet wordnet
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/benediktalkin/imagenetutils
- Owner: BenediktAlkin
- License: mit
- Created: 2023-10-18T21:09:05.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2023-10-19T15:25:57.000Z (about 2 years ago)
- Last Synced: 2025-01-25T07:42:33.884Z (9 months ago)
- Topics: computervision, imagenet, wordnet
- Language: Python
- Homepage:
- Size: 93.8 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# ImageNetUtils
Utilities for analyzing ImageNet classes.
# Setup
`pip install git+https://github.com/BenediktAlkin/ImageNetUtils`
# Examples
## Class index to class name
``` python
from imagenet_utils import index_to_names, index_to_shortest_name
print(index_to_names(288))
# ['leopard', 'Panthera pardus']
print(index_to_shortest_name(288))
# leopard
```
## Class index to WordNet id
``` python
from imagenet_utils import index_to_wordnetid
print(index_to_wordnetid(5))
# n01496331
```
## Classes of a high-level WordNet id
A good visualization of the ImageNet hierarchy can be found [here](https://observablehq.com/@mbostock/imagenet-hierarchy).
``` python
from imagenet_utils import wordnetid_to_leafwordnetids, wordnetid_to_leafnames, wordnetid_to_leafindices
# n02120997 is "feline"
print(wordnetid_to_leafwordnetids("n02120997"))
# ['n02123045', 'n02123159', 'n02123394', 'n02123597', 'n02124075', 'n02125311', 'n02127052', 'n02128385', 'n02128757', 'n02128925', 'n02129165', 'n02129604', 'n02130308']
print(wordnetid_to_leafnames("n02120997"))
# ['tabby', 'tiger cat', 'Persian cat', 'Siamese', 'Egyptian cat', 'puma', 'lynx', 'leopard', 'ounce', 'jaguar', 'lion', 'tiger', 'chetah']
print(wordnetid_to_leafindices("n02120997"))
# [281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293]
```
## Class name to WordNet id
``` python
from imagenet_utils import index_to_wordnetid
print(name_to_wordnetid("feline"))
# n02120997
```
## Word