Update plot.py (ultralytics#7654)

fix bug in issue ultralytics#7650
fhkiel-mlaip · May 1, 2022 · 8462001 · 8462001
1 parent 3836434
commit 8462001
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/plots.py b/utils/plots.py
@@ -426,7 +426,7 @@ def plot_results(file='path/to/results.csv', dir=''):
             s = [x.strip() for x in data.columns]
             x = data.values[:, 0]
             for i, j in enumerate([1, 2, 3, 4, 5, 8, 9, 10, 6, 7]):
-                y = data.values[:, j]
+                y = data.values[:, j].astype('float')
                 # y[y == 0] = np.nan  # don't show zero values
                 ax[i].plot(x, y, marker='.', label=f.stem, linewidth=2, markersize=8)
                 ax[i].set_title(s[j], fontsize=12)