【英偉達發(fā)布數(shù)據(jù)增強庫和圖像解碼庫，可支持多種數(shù)據(jù)格式的可移植訓(xùn)練工作流】
又有新的開源庫來了，英偉達出品，數(shù)量×2。
正在舉行的CVPR2018大會上，英偉達開源了數(shù)據(jù)增強庫NVIDIADALI和圖像解碼庫NVIDIAnvJPEG。
兩個新庫都是什么畫風？簡而言之，NVIDIADALI是一個GPU加速的數(shù)據(jù)增強和圖像加載庫，為優(yōu)化深度學習框架數(shù)據(jù)pipeline而設(shè)計，而NVIDIAnvJPEG是用于JPEG解碼的高性能GPU加速庫。
具體效果如何？我們繼續(xù)往下看。
詳細介紹
一般來說，計算機視覺應(yīng)用里有多階段預(yù)處理數(shù)據(jù)pipeline，包括從磁盤加載和提取數(shù)據(jù)、解碼、裁剪和調(diào)整大小、顏色和空間轉(zhuǎn)換和格式轉(zhuǎn)換等數(shù)據(jù)密集型步驟。
通過DALI，研究人員可利用MXNet、TensorFlow和PyTorch在AWSP38GPU實例或VoltaGPU的DGX-1系統(tǒng)上擴展訓(xùn)練性能。
綜合來看，DALI的方便之處在于：
能與MxNet、TensorFlow和PyTorch等簡單的框架集成
具有支持多種數(shù)據(jù)格式的可移植訓(xùn)練工作流，如JPEG、原始格式、LMDB、RecordIO和TFRecord等
具有可配置圖形和自定義operator的自定義數(shù)據(jù)pipeline
當DALI進行高性能GPU加速解碼時，就需要nvJPEG登場了。這個GPU加速庫支持單個和批處理圖像的解碼、顏色空間轉(zhuǎn)換、多相位解碼，以及CPU和GPU混合解碼。
與只使用CPU解碼的應(yīng)用程序相比，用nvJPEG進行解碼的吞吐量更高，延遲低了。
網(wǎng)站客服
粵公網(wǎng)安備 44030402000946號