Update test.py
Browse files
test.py
CHANGED
|
@@ -162,7 +162,6 @@ def test(data,
|
|
| 162 |
|
| 163 |
# Plot images
|
| 164 |
if batch_i < 1:
|
| 165 |
-
|
| 166 |
f = Path(save_dir) / ('test_batch%g_gt.jpg' % batch_i) # filename
|
| 167 |
plot_images(img, targets, paths, str(f), names) # ground truth
|
| 168 |
f = Path(save_dir) / ('test_batch%g_pred.jpg' % batch_i)
|
|
|
|
| 162 |
|
| 163 |
# Plot images
|
| 164 |
if batch_i < 1:
|
|
|
|
| 165 |
f = Path(save_dir) / ('test_batch%g_gt.jpg' % batch_i) # filename
|
| 166 |
plot_images(img, targets, paths, str(f), names) # ground truth
|
| 167 |
f = Path(save_dir) / ('test_batch%g_pred.jpg' % batch_i)
|