mirror of https://github.com/WongKinYiu/yolov7.git
84 lines
1.3 KiB
Python
84 lines
1.3 KiB
Python
from enum import Enum
|
|
|
|
class COCOLabels(Enum):
|
|
PERSON = 0
|
|
BICYCLE = 1
|
|
CAR = 2
|
|
MOTORBIKE = 3
|
|
AEROPLANE = 4
|
|
BUS = 5
|
|
TRAIN = 6
|
|
TRUCK = 7
|
|
BOAT = 8
|
|
TRAFFIC_LIGHT = 9
|
|
FIRE_HYDRANT = 10
|
|
STOP_SIGN = 11
|
|
PARKING_METER = 12
|
|
BENCH = 13
|
|
BIRD = 14
|
|
CAT = 15
|
|
DOG = 16
|
|
HORSE = 17
|
|
SHEEP = 18
|
|
COW = 19
|
|
ELEPHANT = 20
|
|
BEAR = 21
|
|
ZEBRA = 22
|
|
GIRAFFE = 23
|
|
BACKPACK = 24
|
|
UMBRELLA = 25
|
|
HANDBAG = 26
|
|
TIE = 27
|
|
SUITCASE = 28
|
|
FRISBEE = 29
|
|
SKIS = 30
|
|
SNOWBOARD = 31
|
|
SPORTS_BALL = 32
|
|
KITE = 33
|
|
BASEBALL_BAT = 34
|
|
BASEBALL_GLOVE = 35
|
|
SKATEBOARD = 36
|
|
SURFBOARD = 37
|
|
TENNIS_RACKET = 38
|
|
BOTTLE = 39
|
|
WINE_GLASS = 40
|
|
CUP = 41
|
|
FORK = 42
|
|
KNIFE = 43
|
|
SPOON = 44
|
|
BOWL = 45
|
|
BANANA = 46
|
|
APPLE = 47
|
|
SANDWICH = 48
|
|
ORANGE = 49
|
|
BROCCOLI = 50
|
|
CARROT = 51
|
|
HOT_DOG = 52
|
|
PIZZA = 53
|
|
DONUT = 54
|
|
CAKE = 55
|
|
CHAIR = 56
|
|
SOFA = 57
|
|
POTTEDPLANT = 58
|
|
BED = 59
|
|
DININGTABLE = 60
|
|
TOILET = 61
|
|
TVMONITOR = 62
|
|
LAPTOP = 63
|
|
MOUSE = 64
|
|
REMOTE = 65
|
|
KEYBOARD = 66
|
|
CELL_PHONE = 67
|
|
MICROWAVE = 68
|
|
OVEN = 69
|
|
TOASTER = 70
|
|
SINK = 71
|
|
REFRIGERATOR = 72
|
|
BOOK = 73
|
|
CLOCK = 74
|
|
VASE = 75
|
|
SCISSORS = 76
|
|
TEDDY_BEAR = 77
|
|
HAIR_DRIER = 78
|
|
TOOTHBRUSH = 79
|