Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
52c0570
·
unverified ·
1 Parent(s): abfcf9e

Update Colors() (#3046)

Browse files

* Update Colors()

* update colors

* update colors

Files changed (1) hide show
  1. utils/plots.py +4 -1
utils/plots.py CHANGED
@@ -28,7 +28,10 @@ matplotlib.use('Agg') # for writing to files only
28
  class Colors:
29
  # Ultralytics color palette https://ultralytics.com/
30
  def __init__(self):
31
- self.palette = [self.hex2rgb(c) for c in matplotlib.colors.TABLEAU_COLORS.values()]
 
 
 
32
  self.n = len(self.palette)
33
 
34
  def __call__(self, i, bgr=False):
 
28
  class Colors:
29
  # Ultralytics color palette https://ultralytics.com/
30
  def __init__(self):
31
+ # hex = matplotlib.colors.TABLEAU_COLORS.values()
32
+ hex = ('FF3838', 'FF9D97', 'FF701F', 'FFB21D', 'CFD231', '48F90A', '92CC17', '3DDB86', '1A9334', '00D4BB',
33
+ '2C99A8', '00C2FF', '344593', '6473FF', '0018EC', '8438FF', '520085', 'CB38FF', 'FF95C8', 'FF37C7')
34
+ self.palette = [self.hex2rgb('#' + c) for c in hex]
35
  self.n = len(self.palette)
36
 
37
  def __call__(self, i, bgr=False):