Commit
·
9dbaa1b
1
Parent(s):
83dc540
Update plot_results()
Browse files- utils/general.py +2 -2
utils/general.py
CHANGED
@@ -143,7 +143,7 @@ def check_dataset(dict):
|
|
143 |
if not all(os.path.exists(x) for x in val):
|
144 |
print('\nWARNING: Dataset not found, nonexistant paths: %s' % [*val])
|
145 |
if s and len(s): # download script
|
146 |
-
print('
|
147 |
if s.startswith('http') and s.endswith('.zip'): # URL
|
148 |
f = Path(s).name # filename
|
149 |
if platform.system() == 'Darwin': # avoid MacOS python requests certificate error
|
@@ -1263,7 +1263,7 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(),
|
|
1263 |
y[y == 0] = np.nan # dont show zero loss values
|
1264 |
# y /= y[0] # normalize
|
1265 |
label = labels[fi] if len(labels) else Path(f).stem
|
1266 |
-
ax[i].plot(x, y, marker='.', label=label, linewidth=
|
1267 |
ax[i].set_title(s[i])
|
1268 |
# if i in [5, 6, 7]: # share train and val loss y axes
|
1269 |
# ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])
|
|
|
143 |
if not all(os.path.exists(x) for x in val):
|
144 |
print('\nWARNING: Dataset not found, nonexistant paths: %s' % [*val])
|
145 |
if s and len(s): # download script
|
146 |
+
print('Downloading %s ...' % s)
|
147 |
if s.startswith('http') and s.endswith('.zip'): # URL
|
148 |
f = Path(s).name # filename
|
149 |
if platform.system() == 'Darwin': # avoid MacOS python requests certificate error
|
|
|
1263 |
y[y == 0] = np.nan # dont show zero loss values
|
1264 |
# y /= y[0] # normalize
|
1265 |
label = labels[fi] if len(labels) else Path(f).stem
|
1266 |
+
ax[i].plot(x, y, marker='.', label=label, linewidth=1, markersize=6)
|
1267 |
ax[i].set_title(s[i])
|
1268 |
# if i in [5, 6, 7]: # share train and val loss y axes
|
1269 |
# ax[i].get_shared_y_axes().join(ax[i], ax[i - 5])
|