Skip to content

Contains code to evaluate the WordNet synset tree and merge classes of ImageNet based on this information

License

Notifications You must be signed in to change notification settings

jenuk/imagenet_metaclasses

Repository files navigation

Create Imagenet Metaclasses

Use parse.py to create metaclasses to ImageNet by automatically merging classes along the paths defined by WordNet. Requires the files data.noun from WordNet and either imagenet_synsets.txt (ls ILSVRC2012_train > imagenet_synsets.txt) or imagenet_counts (use count_imagenet.py and point to the ImageNet Folder), all three are provided here.

Outputs the following files:

  • out.yaml the merged version of the synset tree.
  • class_dict.yaml a dictionary mapping the old classes to the new merged classes.
  • leaf_info.txt the resulting classes ordered by size.

Here is an example of the resulting sizes for a "bottom-up" merge

About

Contains code to evaluate the WordNet synset tree and merge classes of ImageNet based on this information

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages