mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update plot_results()

Browse Source
This commit is contained in:
Glenn Jocher 2020-08-22 15:32:05 -07:00
parent 83dc540b1d
commit 9dbaa1bbbe
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/general.py
View File

@ -143,7 +143,7 @@ def check_dataset(dict):
if not all(os.path.exists(x) for x in val):
print('\nWARNING: Dataset not found, nonexistant paths: %s' % [*val])
if s and len(s): # download script
print('Attempting autodownload from: %s' % s)
print('Downloading %s ...' % s)
if s.startswith('http') and s.endswith('.zip'): # URL
f = Path(s).name # filename
if platform.system() == 'Darwin': # avoid MacOS python requests certificate error
@ -1263,7 +1263,7 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(),
y[y == 0] = np.nan # dont show zero loss values
# y /= y[0] # normalize
label = labels[fi] if len(labels) else Path(f).stem
ax[i].plot(x, y, marker='.', label=label, linewidth=2, markersize=8)
ax[i].plot(x, y, marker='.', label=label, linewidth=1, markersize=6)
ax[i].set_title(s[i])
# if i in [5, 6, 7]: # share train and val loss y axes
# ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])