From ecbe041041c78ac84d9d9222fffd2b7bb3bc52d6 Mon Sep 17 00:00:00 2001
From: Ruben van de Ven <git@rubenvandeven.com>
Date: Thu, 13 Apr 2023 13:11:58 +0200
Subject: [PATCH] tqdm.auto to support notebook

---
 track.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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