coarse_to_fine_cifar.py

# Let's map between the fine and coarse categories.
cifar100Dir = '/data/cifar-100-python/' # modify this to your own directory...
import cPickle as pickle
import os
meta = pickle.load(open(os.path.join(cifar100Dir,'meta')))
test = pickle.load(open(os.path.join(cifar100Dir,'test')))
test_fine_labels = test['fine_labels']
test_coarse_labels = test['coarse_labels']
fine_to_coarse = dict(set((zip(test_fine_labels,test_coarse_labels))))


fine_label_names = meta['fine_label_names']
coarse_label_names = meta['coarse_label_names']
coarse_to_fine = {}
for k,v in fine_to_coarse.iteritems():
    #print fine_label_names[k],'-->',coarse_label_names[v]
    if v not in coarse_to_fine:
        coarse_to_fine[v] = [k]
    else:
        coarse_to_fine[v].append(k)
print 'coarse-->fine mapping'        
print '---------------------'
for i,j in coarse_to_fine.iteritems():
    print coarse_label_names[i],':',
    for k in j:
        print fine_label_names[k],',',
    print