Make CONFIG_DIR configurable per environment variable (#4727)

positive666 · Sep 10, 2021 · a9d388a · a9d388a
1 parent d44716d
commit a9d388a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils/plots.py b/utils/plots.py
@@ -4,6 +4,7 @@
 """
 
 import math
+import os
 from copy import copy
 from pathlib import Path
 
@@ -20,6 +21,7 @@
 from utils.metrics import fitness
 
 # Settings
+CONFIG_DIR = Path(os.getenv('YOLOV5_CONFIG_DIR') or user_config_dir())  # Ultralytics settings dir
 matplotlib.rc('font', **{'size': 11})
 matplotlib.use('Agg')  # for writing to files only