diff --git a/utils/plots.py b/utils/plots.py index 1ed88ea7c832..141a9ac2b09e 100644 --- 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