From a9d388a3ce5f0b273d4dc2884f62389d658a4e66 Mon Sep 17 00:00:00 2001
From: positive666 <286040359@qq.com>
Date: Fri, 10 Sep 2021 17:34:37 +0800
Subject: [PATCH] Make CONFIG_DIR configurable per environment variable (#4727)

---
 utils/plots.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/utils/plots.py b/utils/plots.py
index d8a561a7..e50e1f46 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,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