瀏覽代碼

change legend from low_right to low_left for score-recall curve

FlyingQianMM 5 年之前
父節點
當前提交
5925df7582
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/cv/models/utils/visualize.py

+ 1 - 1
paddlex/cv/models/utils/visualize.py

@@ -278,7 +278,7 @@ def draw_pr_curve(eval_details_file=None,
             color = [float(c) / 255 for c in color]
             color = [float(c) / 255 for c in color]
             color.append(0.75)
             color.append(0.75)
             plt.plot(x, sr_array, color=color, label=nm, linewidth=1)
             plt.plot(x, sr_array, color=color, label=nm, linewidth=1)
-        plt.legend(loc="lower right", fontsize=5)
+        plt.legend(loc="lower left", fontsize=5)
         plt.savefig(
         plt.savefig(
             os.path.join(save_dir, "./{}_pr_curve(iou-{}).png".format(
             os.path.join(save_dir, "./{}_pr_curve(iou-{}).png".format(
                 style, iou_thresh)),
                 style, iou_thresh)),