Minor fix for Class initialization at Inferer

ZHEQIUSHUI · Jun 22, 2022 · 6baec25 · 6baec25
1 parent ba96a20
commit 6baec25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolov6/core/inferer.py b/yolov6/core/inferer.py
@@ -18,7 +18,7 @@
 from yolov6.utils.nms import non_max_suppression
 
 
-class Inferer():
+class Inferer:
     def __init__(self, source, weights, device, yaml, img_size, half):
         import glob
         from yolov6.data.datasets import IMG_FORMATS
@@ -193,4 +193,4 @@ def generate_colors(i, bgr=False):
             palette.append(tuple(int(h[1 + i:1 + i + 2], 16) for i in (0, 2, 4)))
         num = len(palette)
         color = palette[int(i) % num]
-        return (color[2], color[1], color[0]) if bgr else color
+        return (color[2], color[1], color[0]) if bgr else color