Remove Matplotlib agg backend (#499)

* Remove Matplotlib agg backend https://stackoverflow.com/questions/56656777/userwarning-matplotlib-is-currently-using-agg-which-is-a-non-gui-backend-so
zhiqwang · Jul 31, 2023 · a2790eb · a2790eb
1 parent 12809ca
commit a2790eb
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/yolort/v5/utils/plots.py b/yolort/v5/utils/plots.py
@@ -36,8 +36,6 @@
 # Settings
 CONFIG_DIR = user_config_dir()  # Ultralytics settings dir
 RANK = int(os.getenv("RANK", -1))
-matplotlib.rc("font", **{"size": 11})
-matplotlib.use("Agg")  # for writing to files only
 
 
 class Colors:
@@ -446,7 +444,6 @@ def plot_labels(labels, names=(), save_dir=Path("")):
             ax[a].spines[s].set_visible(False)
 
     plt.savefig(save_dir / "labels.jpg", dpi=200)
-    matplotlib.use("Agg")
     plt.close()