多線程處理腳本
在處理數據集的時候,往往是一張一張循環處理,如果數據集太大,處理時間就會很長。我們可以使用多線程處理。
多線程的模板直接用mmcv庫里面的一個函數就行了
參考如下:
https://github.com/open-mmlab/mmcv/blob/master/docs_zh_CN/understand_mmcv/utils.md
比如我要寫一個cp操作,把數據集按照一定格式復制一遍。
import os import mmcv def func_cp(item): img_path, cp_path = item cmd = "cp " + img_path + " " + cp_path os.system(cmd) items_cp = [] for ... : items_cp.append((img_path, cp_path)) mmcv.track_parallel_progress(func_cp, items_cp, 32)

浙公網安備 33010602011771號