diff --git a/tools/program.py b/tools/program.py index c22bf18b991a8aed6d47a1ea242aa3b7bb02aacc..33bd35c23d0b6676edfb9c537e497eb68b828c4a 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 @@ -335,8 +336,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()