Update val.py

ultralytics · Nov 3, 2021 · 97124a3 · 97124a3
1 parent 94b5186
commit 97124a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/val.py b/val.py
@@ -160,8 +160,8 @@ def run(data,
     dt, p, r, f1, mp, mr, map50, map = [0.0, 0.0, 0.0], 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0
     loss = torch.zeros(3, device=device)
     jdict, stats, ap, ap_class = [], [], [], []
-    pbar = tqdm(enumerate(dataloader), desc=s, bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}')  # progress bar
-    for batch_i, (img, targets, paths, shapes) in pbar:
+    pbar = tqdm(dataloader, desc=s, bar_format='{l_bar}{bar:10}{r_bar}{bar:-10b}')  # progress bar
+    for batch_i, (img, targets, paths, shapes) in enumerate(pbar):
         t1 = time_sync()
         img = img.to(device, non_blocking=True)
         img = img.half() if half else img.float()  # uint8 to fp16/32