Unverified 提交 deb434ae authored 作者: JD Costa's avatar JD Costa 提交者: GitHub

Make CONFIG_DIR configurable per environment variable (#4727)

上级 0d8a1842
...@@ -4,6 +4,7 @@ Plotting utils ...@@ -4,6 +4,7 @@ Plotting utils
""" """
import math import math
import os
from copy import copy from copy import copy
from pathlib import Path from pathlib import Path
...@@ -20,7 +21,7 @@ from utils.general import user_config_dir, is_ascii, xywh2xyxy, xyxy2xywh ...@@ -20,7 +21,7 @@ from utils.general import user_config_dir, is_ascii, xywh2xyxy, xyxy2xywh
from utils.metrics import fitness from utils.metrics import fitness
# Settings # 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.rc('font', **{'size': 11})
matplotlib.use('Agg') # for writing to files only matplotlib.use('Agg') # for writing to files only
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论