diff --git a/utils/utils.py b/utils/utils.py index a3795e64..26a450f5 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -520,7 +520,7 @@ def plot_results(start=0, stop=0): # from utils.utils import *; plot_results() fig = plt.figure(figsize=(14, 7)) s = ['X + Y', 'Width + Height', 'Confidence', 'Classification', 'Train Loss', 'Precision', 'Recall', 'mAP', 'F1', 'Test Loss'] - for f in sorted(glob.glob('results*.txt')): + for f in sorted(glob.glob('results*.txt') + glob.glob('../../Downloads/results*.txt')): results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 9, 10, 11, 12, 13]).T n = results.shape[1] # number of rows x = range(start, min(stop, n) if stop else n)