perf(draw): 中文乱码；每次重新绘制新的图片

Author: zjZSTU

pynet/vision/draw.py

@@ -6,6 +6,9 @@
 
 import matplotlib.pyplot as plt
 
+plt.rcParams['font.sans-serif'] = ['simhei']  # 用来正常显示中文标签
+plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号
+
 
 class Draw(object):
 
@@ -14,6 +17,7 @@ def __call__(self, values, title='损失图', xlabel='迭代/次', ylabel='损
 
     def forward(self, values, title='损失图', xlabel='迭代/次', ylabel='损失值', save_path='./loss.png'):
         assert isinstance(values, list)
+        f = plt.figure()
         plt.title(title)
         plt.ylabel(ylabel)
         plt.xlabel(xlabel)
@@ -26,6 +30,7 @@ def multi_plot(self, values_list, labels_list, title='损失图', xlabel='迭代
         assert isinstance(labels_list, tuple)
         assert len(values_list) == len(labels_list)
 
+        f = plt.figure()
         plt.title(title)
         plt.ylabel(ylabel)
         plt.xlabel(xlabel)