1. 程式人生 > >pytorch 從頭開始faster-rcnn(零):使用到的函式

pytorch 從頭開始faster-rcnn(零):使用到的函式

本篇部落格是在寫faster-rcnn遇到的沒見過的函式,所以這篇部落格隨著程式碼的編寫不定期更新。

1.tqdm

tqdm在阿拉伯語中的意思是進展。tqdm可以在長迴圈中新增一個進度提示資訊,使用者只需要封裝任意的迭代器 tqdm(iterator),是一個快速、擴充套件性強的進度條工具庫。

from tqdm import tqdm
from time import sleep

for i in tqdm(range(1000)):
    sleep(0.01)

使用方法二: trange

trange(i) 是 tqdm(range(i)) 的簡單寫法

from tqdm import trange
for i in trange(100):
    #do something
    pass

使用方法三: 手動方法

在for迴圈外部初始化tqdm,可以列印其他資訊

pbar = tqdm(["a", "b", "c", "d"])
for char in pbar:
    pbar.set_description("Processing %s" % char)

2.