fix: stroke color when not specified
Browse files- 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] =
|
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 |
|