Make CONFIG_DIR configurable per environment variable

ultralytics · Sep 9, 2021 · e63ab42 · e63ab42
1 parent 0d8a184
commit e63ab42
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', user_config_dir()))  # Ultralytics settings dir
 matplotlib.rc('font', **{'size': 11})
 matplotlib.use('Agg')  # for writing to files only