基于Yolov7的路面病害检测代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

84 lines
1.3KB

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