项目场景:
在使用tqdm显示dataloader进度时,出现多余的0%进度条
代码如下
for i, data in enumerate(tqdm(data_generator(TifArray), position = 0, total = len(TifArray)), ):
问题描述
运行时显示如下
裁剪行列数为: 6 , 6 100%|██████████| 6/6 [00:12<00:00, 2.13s/it] 0%| | 0/6 [00:00<?, ?it/s] ------------ sample3_clip1_s002.png ------------ 裁剪行列数为: 6 , 6 100%|██████████| 6/6 [00:11<00:00, 1.99s/it] 0%| | 0/6 [00:00<?, ?it/s]
解决方案:
参考大佬评论,tqdm加入参数file=sys.stdout
代码如下:
for i, data in enumerate(tqdm(data_generator(TifArray), position = 0, total = len(TifArray),file=sys.stdout)):
运行结果如下
------------ sample3_clip1_s002.png ------------
裁剪行列数为: 6 , 6
100%|██████████| 6/6 [00:11<00:00, 1.94s/it]------------ sample3_clip1_s003.png ------------
裁剪行列数为: 6 , 6
100%|██████████| 6/6 [00:11<00:00, 1.98s/it]