GIoU to default

2019-07-07 23:53:56 +02:00 · 2019-07-07 23:53:56 +02:00 · 7a2d356297
parent af3c5d0e35
commit 7a2d356297
1 changed files with 2 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -629,7 +629,8 @@ def plot_images(imgs, targets, paths=None, fname='images.jpg'):
        plt.plot(boxes[[0, 2, 2, 0, 0]], boxes[[1, 1, 3, 3, 1]], '.-')
        plt.axis('off')
        if paths is not None:
-            plt.title(Path(paths[i]).name, fontdict={'size': 8})
+            s = Path(paths[i]).name
+            plt.title(s[:min(len(s), 40)], fontdict={'size': 8})  # limit to 40 characters
    fig.tight_layout()
    fig.savefig(fname, dpi=300)
    plt.close()