Update plot.py (#7654)

fix bug in issue #7650
2022-05-01 11:01:21 +08:00 · 2022-05-01 11:01:21 +08:00 · 5743deb9df
parent 488fb0a36a
commit 5743deb9df
1 changed files with 1 additions and 1 deletions
--- 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)