bai 3 years ago
parent
commit
643616f2e8
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test.py

+ 1
- 1
test.py View File

@@ -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()


Loading…
Cancel
Save