Make CONFIG_DIR configurable per environment variable (ultralytics#4727)

fhkiel-mlaip · Sep 9, 2021 · c414f76 · c414f76
1 parent bded191
commit c414f76
Showing 1 changed file with 2 additions and 1 deletion.
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,7 +21,7 @@
 from utils.metrics import fitness
 
 # Settings
-CONFIG_DIR = user_config_dir()  # Ultralytics settings dir
+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