mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
Update test.py
This commit is contained in:
parent
c3d3e6b776
commit
22ab1c295f
1
test.py
1
test.py
@ -162,7 +162,6 @@ def test(data,
|
|||||||
|
|
||||||
# Plot images
|
# Plot images
|
||||||
if batch_i < 1:
|
if batch_i < 1:
|
||||||
|
|
||||||
f = Path(save_dir) / ('test_batch%g_gt.jpg' % batch_i) # filename
|
f = Path(save_dir) / ('test_batch%g_gt.jpg' % batch_i) # filename
|
||||||
plot_images(img, targets, paths, str(f), names) # ground truth
|
plot_images(img, targets, paths, str(f), names) # ground truth
|
||||||
f = Path(save_dir) / ('test_batch%g_pred.jpg' % batch_i)
|
f = Path(save_dir) / ('test_batch%g_pred.jpg' % batch_i)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user