Update plots.py with yaml.SafeLoader (ultralytics#1968)

taicaile · Jan 18, 2021 · a06670d · a06670d
1 parent 38ebcb0
commit a06670d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/plots.py b/utils/plots.py
@@ -301,7 +301,7 @@ def plot_labels(labels, save_dir=Path(''), loggers=None):
 def plot_evolution(yaml_file='data/hyp.finetune.yaml'):  # from utils.plots import *; plot_evolution()
     # Plot hyperparameter evolution results in evolve.txt
     with open(yaml_file) as f:
-        hyp = yaml.load(f, Loader=yaml.FullLoader)
+        hyp = yaml.load(f, Loader=yaml.SafeLoader)
     x = np.loadtxt('evolve.txt', ndmin=2)
     f = fitness(x)
     # weights = (f - f.min()) ** 2  # for weighted results