|
|
|
@@ -207,7 +207,7 @@ def test(data, |
|
|
|
# Append statistics (correct, conf, pcls, tcls) |
|
|
|
stats.append((correct.cpu(), pred[:, 4].cpu(), pred[:, 5].cpu(), tcls)) |
|
|
|
|
|
|
|
# Plot images# file add images and |
|
|
|
# Plot images# file add images and video |
|
|
|
if plots and batch_i < 3: |
|
|
|
f = save_dir / f'test_batch{batch_i}_labels.jpg' # labels |
|
|
|
Thread(target=plot_images, args=(img, targets, paths, f, names), daemon=True).start() |
|
|
|
|