Batch plot the log files exported from VisualDL using Matplotlib.
VisualDL | VDLdraw |
---|---|
It can be installed through pip
:
pip install vdldraw
Online experiences in AI Studio (Chinese).
from vdldraw import get_tags, vdl_draw_folder
## 1.View available tags
get_tags(log_folder)
# Output like this:
# ['Evaluate/F1', 'Train/loss', 'Evaluate/mIoU', 'Evaluate/Acc']
## 2.Make up tags to be displayed
tag_list = ['Evaluate/mIoU', 'Evaluate/Acc']
## 3.Save image
vdl_draw_folder(log_folder, tag_list, save_folder="output", mplstyle=None)
# Args:
# log_folder (str): Folder path where logs are saved.
# tag_list (list): List of tags to be displayed.
# save_folder (str, option): Folder path to save pictures. Default: "output".
# mplstyle (str/path): Matplotlib style system. Default: None(Styles defined using VDLdraw).
Colormap reference antv's contrasting orange, 20 colors in total.
If you need use some styles, you could use SciencePlots. This repo has Matplotlib styles to format your figures for scientific papers, presentations and theses.