diff --git a/tools/program.py b/tools/program.py index 0aa3307e38982985519e3156214c0b4991ef54bc..db8e44df360d8d255406a13ae697f598d7a96a3a 100755 --- a/tools/program.py +++ b/tools/program.py @@ -18,6 +18,7 @@ from __future__ import print_function import os import sys +import platform import yaml import time import shutil @@ -338,8 +339,10 @@ def eval(model, valid_dataloader, post_process_class, eval_class, total_frame = 0.0 total_time = 0.0 pbar = tqdm(total=len(valid_dataloader), desc='eval model:') + max_iter = len(valid_dataloader) - 1 if platform.system( + ) == "Windows" else len(valid_dataloader) for idx, batch in enumerate(valid_dataloader): - if idx >= len(valid_dataloader): + if idx >= max_iter: break images = batch[0] start = time.time()