import numpy as np

cnames = {
    'background'      : '0',
    'apple'           : '1',
    'ball'            : '2',
    'banana'          : '3',
    'bell_pepper'     : '4',
    'binder'          : '5',
    'bowl'            : '6',
    'cereal_box'      : '7',
    'coffee_mug'      : '8',
    'flashlight'      : '9',
    'food_bag'        : '10',
    'food_box'        : '11',
    'food_can'        : '12',
    'glue_stick'      : '13',
    'hand_towel'      : '14',
    'instant_noodles' : '15',
    'keyboard'        : '16',
    'kleenex'         : '17',
    'lemon'           : '18',
    'lime'            : '19',
    'marker'          : '20',
    'orange'          : '21',
    'peach'           : '22',
    'pear'            : '23',
    'potato'          : '24',
    'shampoo'         : '25',
    'soda_can'        : '26',
    'sponge'          : '27',
    'stapler'         : '28',
    'tomato'          : '29',
    'toothpaste'      : '30',
    'unknown'         : '31'
    }

colors = {
    'b': np.array([0.0, 0.0, 1.0])*255,
    'g': np.array([0.0, 0.5, 0.0])*255,
    'r': np.array([1.0, 0.0, 0.0])*255,
    'c': np.array([0.0, 0.75, 0.75])*255,
    'm': np.array([0.75, 0, 0.75])*255,
    'y': np.array([0.75, 0.75, 0])*255,
     'w': np.array([1.0, 1.0, 1.0])*255,
}

colors_list = list(colors.values())



subnames = {'background': 0,
 'apple_1': 1,
 'apple_2': 2,
 'ball_1': 3,
 'ball_2': 4,
 'ball_3': 5,
 'banana_1': 6,
 'banana_2': 7,
 'bell_pepper_1': 8,
 'binder_1': 9,
 'bowl_1': 10,
 'cereal_box_1': 11,
 'cereal_box_3': 12,
 'cereal_box_4': 13,
 'cereal_box_5': 14,
 'coffee_mug_1': 15,
 'coffee_mug_2': 16,
 'flashlight_1': 17,
 'food_bag_2': 18,
 'food_bag_3': 19,
 'food_bag_4': 20,
 'food_box_1': 21,
 'food_box_2': 22,
 'food_box_3': 23,
 'food_can_1': 24,
 'food_can_2': 25,
 'food_can_3': 26,
 'glue_stick_1': 27,
 'hand_towel_1': 28,
 'hand_towel_2': 29,
 'hand_towel_3': 30,
 'instant_noodles_1': 31,
 'instant_noodles_2': 32,
 'keyboard_1': 33,
 'keyboard_2': 34,
 'kleenex_1': 35,
 'kleenex_2': 36,
 'kleenex_3': 37,
 'lemon_1': 38,
 'lemon_2': 39,
 'lime_1': 40,
 'lime_2': 41,
 'marker_1': 42,
 'marker_2': 43,
 'marker_3': 44,
 'orange_1': 45,
 'orange_2': 46,
 'peach_1': 47,
 'peach_2': 48,
 'pear_1': 49,
 'pear_2': 50,
 'potato_1': 51,
 'potato_2': 52,
 'shampoo_1': 53,
 'shampoo_2': 54,
 'shampoo_3': 55,
 'soda_can_1': 56,
 'soda_can_2': 57,
 'sponge_1': 58,
 'sponge_2': 59,
 'sponge_3': 60,
 'stapler_1': 61,
 'stapler_2': 62,
 'tomato_1': 63,
 'toothpaste_1': 64,
 'toothpaste_2': 65,
 'unknown': 66
}

sub_to_class = {0: 0,
 1: 1,
 2: 1,
 3: 2,
 4: 2,
 5: 2,
 6: 3,
 7: 3,
 8: 4,
 9: 5,
 10: 6,
 11: 7,
 12: 7,
 13: 7,
 14: 7,
 15: 8,
 16: 8,
 17: 9,
 18: 10,
 19: 10,
 20: 10,
 21: 11,
 22: 11,
 23: 11,
 24: 12,
 25: 12,
 26: 12,
 27: 13,
 28: 14,
 29: 14,
 30: 14,
 31: 15,
 32: 15,
 33: 16,
 34: 16,
 35: 17,
 36: 17,
 37: 17,
 38: 18,
 39: 18,
 40: 19,
 41: 19,
 42: 20,
 43: 20,
 44: 20,
 45: 21,
 46: 21,
 47: 22,
 48: 22,
 49: 23,
 50: 23,
 51: 24,
 52: 24,
 53: 25,
 54: 25,
 55: 25,
 56: 26,
 57: 26,
 58: 27,
 59: 27,
 60: 27,
 61: 28,
 62: 28,
 63: 29,
 64: 30,
 65: 30,
 66: 31
}