Update plot.py (ultralytics#7654)

fix bug in issue ultralytics#7650
tdhooghe · Jun 10, 2022 · 246d775 · 246d775
1 parent 83c1961
commit 246d775
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)