diff --git a/track.py b/track.py index 26406f0..a7804c7 100644 --- a/track.py +++ b/track.py @@ -5,7 +5,7 @@ import cv2 import logging import argparse -import tqdm +from tqdm.auto import tqdm import motmetrics as mm import torch @@ -85,7 +85,7 @@ def eval_seq(opt, dataloader, data_type, result_filename, save_dir=None, save_im timer = Timer() results = [] frame_id = -1 - for path, img, img0 in tqdm.tqdm(dataloader): + for path, img, img0 in tqdm(dataloader): frame_id += 1 # if frame_id % 20 == 0: # logger.info('Processing frame {} ({:.2f} fps)'.format(frame_id, 1./max(1e-5, timer.average_time)))