ObjTest/class_info.py at main · SSCT-Lab/ObjTest · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
coco_class_dict = {
    0: '0_person.png',
    2: '2_car.png',
    4: '4_airplane.png',
    6: '6_train.png',
    8: '8_boat.png',
    10: '10_fire hydrant.png',
    11: '11_stop sign.png',
    15: '15_cat.png',
    16: '16_dog.png',
    22: '22_zebra.png',
    23: '23_giraffe.png',
    46: '46_banana.png',
    48: '48_sandwich.png',
    49: '49_orange.png',
    53: '53_pizza.png',
    54: '54_donut.png',
    61: '61_toilet.png',
    62: '62_tv.png',
    72: '72_refrigerator.png',
    74: '74_clock.png'
}

openimages_class_dict = {
    110: '110_Snack.png',
    121: '121_cat.png',
    228: '228_door handle.png',
    34: '34_Sunglasses.png',
    404: '404_handbag.png',
    426: '426_Turtle.png',
    433: '433_Footwear.png',
    440: '440_wind glass.png',
    445: '445_dog.png',
    455: '455_Flower.png',
    467: '467_Airplane.png',
    483: '483_Moths and butterflies.png',
    498: '498_Camera.png',
    503: '503_Vegetable.png',
    510: '510_Cabbage.png',
    570: '570_car.png',
    592: '592_Alpaca.png',
    595: '595_Remote control.png',
    68: '68_person.png',
    98: '98_poster.png'
}

voc_class_dict = {
    0: '0_aeroplane.png',
    1: '1_bicycle.png',
    2: '2_bird.png',
    4: '4_bottle.png',
    5: '5_bus.png',
    6: '6_car.png',
    7: '7_cat.png',
    11: '11_dog.png',
    14: '14_person.png',
    18: '18_train.png'
}