Update test.py

5 years ago · 22ab1c295f
parent c3d3e6b776
commit 22ab1c295f
1 changed files with 0 additions and 1 deletions
--- a/test.py
+++ b/test.py
@ -162,7 +162,6 @@ def test(data,
        # Plot images
        if batch_i < 1:
            f = Path(save_dir) / ('test_batch%g_gt.jpg' % batch_i)  # filename
            plot_images(img, targets, paths, str(f), names)  # ground truth
            f = Path(save_dir) / ('test_batch%g_pred.jpg' % batch_i)