mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix wrong condition judgment in analyze_logs.py and prevent empty curve. (#510)

Browse Source 
* [Fix] Analyze logs with single training iteration

* [Fix] Make `plot_curve` with 'accuracy' metric

* [Fix] Assert length of training log records of each epoch is larger than 1

Co-authored-by: Ma Zerun <mzr1996@163.com>

Co-authored-by: Ma Zerun <mzr1996@163.com>
This commit is contained in:
imyhxy 2021-11-03 11:37:51 +08:00 committed by GitHub
parent b486bb23f8
commit abb5a222cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/analysis_tools/analyze_logs.py
View File

@ -57,11 +57,9 @@ def plot_curve(log_dicts, args):
f'{args.json_logs[i]} does not contain metric {metric} ' f'{args.json_logs[i]} does not contain metric {metric} '
f'in train mode') f'in train mode')
if 'mAP' in metric: if any(m in metric for m in ('mAP', 'accuracy')):
xs = np.arange(1, max(epochs) + 1) xs = epochs
ys = [] ys = [log_dict[e][metric] for e in xs]
for epoch in epochs:
ys += log_dict[epoch][metric]
ax = plt.gca() ax = plt.gca()
ax.set_xticks(xs) ax.set_xticks(xs)
plt.xlabel('epoch') plt.xlabel('epoch')
@ -74,6 +72,9 @@ def plot_curve(log_dicts, args):
iters = log_dict[epoch]['iter'] iters = log_dict[epoch]['iter']
if log_dict[epoch]['mode'][-1] == 'val': if log_dict[epoch]['mode'][-1] == 'val':
iters = iters[:-1] iters = iters[:-1]
assert len(iters) > 0, (
'The training log is empty, please try to reduce the '
'interval of log in config file.')
xs.append( xs.append(
np.array(iters) + (epoch - 1) * num_iters_per_epoch) np.array(iters) + (epoch - 1) * num_iters_per_epoch)
ys.append(np.array(log_dict[epoch][metric][:len(iters)])) ys.append(np.array(log_dict[epoch][metric][:len(iters)]))