Python之tqdm

最新推荐文章于 2025-03-18 09:05:50 发布

Nicola-Zhang

最新推荐文章于 2025-03-18 09:05:50 发布

阅读量1w

点赞数 1

分类专栏： dataset

本文链接：https://blog.youkuaiyun.com/yangwangnndd/article/details/94651368

版权

本文介绍了Python库tqdm的用途，它主要用于显示进度条。tqdm使用简单，可以轻松封装迭代器来实现进度提示，适用于长循环中。tqdm在不同环境下无需依赖，开销小，且具有阿拉伯语和西班牙语的有趣含义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python之tqdm

Python之tqdm主要作用是用于显示进度，使用较为简单：

创建进度条；
关闭进度条；

注意比较有意思的一点是关于tqdm来源问题：
Tqdm在阿拉伯语表示进步，在西班牙语中表示我非常爱你。是一个快速，可扩展的Python进度条，可以在Python长循环中添加一个进度提示信息，用户只需要封装任意的迭代器tqdm(iterator)即可完成进度条。相比ProgressBar来说Tqdm的开销非常低，同时Tqdm可以在任何环境中不需要任何依赖运行

def calculate_weigths_labels(dataset, dataloader, num_classes):
    # Create an instance from the data loader
    z = np.zeros((num_classes,))
    # Initialize tqdm
    tqdm_batch = tqdm(dataloader)
    print('Calculating classes weights')
    for sample in tqdm_batch:
        y = sample['label']
        y = y.detach().cpu().numpy()
        mask = (y >= 0) & (y < num_classes)
        labels = y[mask].astype(np.uint8)
        count_l = np.bincount(labels, minlength=num_classes)
        z += count_l
    tqdm_batch.c