gyrojeff commited on
Commit
3b25c65
1 Parent(s): 5a85fd3

fix: stroke color when not specified

Browse files
Files changed (1) hide show
  1. detector/data.py +1 -1
detector/data.py CHANGED
@@ -47,7 +47,7 @@ class FontDataset(Dataset):
47
  out[8] = label.stroke_color[1] / 255.0
48
  out[9] = label.stroke_color[2] / 255.0
49
  else:
50
- out[7:10] = 0.5
51
  out[10] = label.line_spacing / label.image_width
52
  out[11] = label.angle / 180.0 + 0.5
53
 
 
47
  out[8] = label.stroke_color[1] / 255.0
48
  out[9] = label.stroke_color[2] / 255.0
49
  else:
50
+ out[7:10] = out[2:5]
51
  out[10] = label.line_spacing / label.image_width
52
  out[11] = label.angle / 180.0 + 0.5
53