mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add Microsoft FocalNet specific ('ms') ImageNet-22k classifier layout

Browse Source
This commit is contained in:
Ross Wightman 2023-03-18 14:57:34 -07:00
parent 572f05096a
commit 9fcfb8bcc1
4 changed files with 55510 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21841
timm/data/_info/imagenet22k_ms_synsets.txt Normal file
View File

File diff suppressed because it is too large Load Diff

11821
timm/data/_info/imagenet22k_ms_to_12k_indices.txt Normal file
View File

File diff suppressed because it is too large Load Diff

21841
timm/data/_info/imagenet22k_ms_to_22k_indices.txt Normal file
View File

File diff suppressed because it is too large Load Diff

11
timm/data/imagenet_info.py
View File

@ -7,12 +7,14 @@ from typing import Dict, List, Optional, Union
from .dataset_info import DatasetInfo from .dataset_info import DatasetInfo
# NOTE no ambiguity wrt to mapping from # classes to ImageNet subset so far, but likely to change
_NUM_CLASSES_TO_SUBSET = { _NUM_CLASSES_TO_SUBSET = {
1000: 'imagenet-1k', 1000: 'imagenet-1k',
11821: 'imagenet-12k', 11221: 'imagenet-21k-miil', # miil subset of fall11
21841: 'imagenet-22k', 11821: 'imagenet-12k', # timm specific 12k subset of fall11
21843: 'imagenet-21k-goog', 21841: 'imagenet-22k', # as in fall11.tar
11221: 'imagenet-21k-miil', 21842: 'imagenet-22k-ms', # a Microsoft (for FocalNet) remapping of 22k w/ moves ImageNet-1k classes to first 1000
21843: 'imagenet-21k-goog', # Google's ImageNet full has two classes not in fall11
} }
_SUBSETS = { _SUBSETS = {
@ -22,6 +24,7 @@ _SUBSETS = {
'imagenet21k': 'imagenet21k_goog_synsets.txt', 'imagenet21k': 'imagenet21k_goog_synsets.txt',
'imagenet21kgoog': 'imagenet21k_goog_synsets.txt', 'imagenet21kgoog': 'imagenet21k_goog_synsets.txt',
'imagenet21kmiil': 'imagenet21k_miil_synsets.txt', 'imagenet21kmiil': 'imagenet21k_miil_synsets.txt',
'imagenet22kms': 'imagenet22k_ms_synsets.txt',
} }
_LEMMA_FILE = 'imagenet_synset_to_lemma.txt' _LEMMA_FILE = 'imagenet_synset_to_lemma.txt'
_DEFINITION_FILE = 'imagenet_synset_to_definition.txt' _DEFINITION_FILE = 'imagenet_synset_to_definition.txt'