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

skip wrong iter number (#716)

Browse Source
This commit is contained in:
谢昕辰 2021-07-27 19:07:36 +08:00 committed by GitHub
parent 65265931c3
commit 58c9ac0cae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tools/analyze_logs.py
View File

@ -30,6 +30,9 @@ def plot_curve(log_dicts, args):
plot_epochs = [] plot_epochs = []
plot_iters = [] plot_iters = []
plot_values = [] plot_values = []
# In some log files, iters number is not correct, `pre_iter` is
# used to prevent generate wrong lines.
pre_iter = -1
for epoch in epochs: for epoch in epochs:
epoch_logs = log_dict[epoch] epoch_logs = log_dict[epoch]
if metric not in epoch_logs.keys(): if metric not in epoch_logs.keys():
@ -39,6 +42,9 @@ def plot_curve(log_dicts, args):
plot_values.append(epoch_logs[metric][0]) plot_values.append(epoch_logs[metric][0])
else: else:
for idx in range(len(epoch_logs[metric])): for idx in range(len(epoch_logs[metric])):
if pre_iter > epoch_logs['iter'][idx]:
continue
pre_iter = epoch_logs['iter'][idx]
plot_iters.append(epoch_logs['iter'][idx]) plot_iters.append(epoch_logs['iter'][idx])
plot_values.append(epoch_logs[metric][idx]) plot_values.append(epoch_logs[metric][idx])
ax = plt.gca() ax = plt.gca()