diff --git a/test.py b/test.py index 6eca724..4790a1c 100644 --- a/test.py +++ b/test.py @@ -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()