雷鋒網(wǎng) AI 科技評論按：NVIDIA 遷移學習工具包對于深度學習應用開發(fā)人員和數(shù)據(jù)科學家來說是理想的工具，這些開發(fā)人員和數(shù)據(jù)科學家正在為各種行業(yè)垂直領(lǐng)域（如智能視頻分析（IVA）和醫(yī)學成像）尋求更快、更高效的深度學習訓練工作流程。

許多應用開發(fā)者和數(shù)據(jù)科學家都正在為智能視頻分析（IVA）、醫(yī)療影像等各種垂直領(lǐng)域?qū)ふ腋?、更高效的深度學習訓練工作流程。NVIDIA 近期就發(fā)布了一個遷移學習工具包（ NVIDIA Transfer Learning Toolkit），希望可以成為他們的理想工具。雷鋒網(wǎng) AI  科技評論編譯。

遷移學習工具包通過允許開發(fā)人員微調(diào) NVIDIA 提供的特定領(lǐng)域的預訓練模型來抽象和加速深度學習訓練，而不是從頭開始花大量時間來從零開始（說明一下，「從零開始」這個詞很重要，它是遷移學習最關(guān)鍵的特點）構(gòu)建深層神經(jīng)網(wǎng)絡。預訓練模型加速了開發(fā)人員的深度學習訓練過程，并且減少了大規(guī)模數(shù)據(jù)收集、標記和從零開始訓練模型相關(guān)的成本。

「遷移學習」意味著可以從現(xiàn)有神經(jīng)網(wǎng)絡中提取已學習特征，并通過從現(xiàn)有神經(jīng)網(wǎng)絡轉(zhuǎn)移權(quán)重來遷移這些已學習特征。NVIDIA 此次發(fā)布的遷移學習工具包是一個基于 Python 的工具包，它使開發(fā)人員能夠利用 NVIDIA 預先訓練好的模型，并為開發(fā)人員提供技術(shù)支持，通過重新訓練讓模型適應新的網(wǎng)絡，來添加他們自己的數(shù)據(jù)以使神經(jīng)網(wǎng)絡更加智能。工具包具備的簡單添加、修剪和重新訓練網(wǎng)絡的能力，提高了深度學習訓練流程的效率和準確性。

主要功能

• 在特定任務上預訓練完畢的、經(jīng)過優(yōu)化的深度神經(jīng)網(wǎng)絡模型已經(jīng)打包在內(nèi)
  

• 用于計算機視覺相關(guān)對象分類、檢測場景的示例應用程序
  

• 在復雜的多 GPU 環(huán)境也可以輕松地模型適配和再訓練
  

• 輕松修改配置文件，從添加新的類和特性到壓縮模型大小都可以支持
  

• 為 IVA 應用程序在 NVIDIA DeepStream SDK 3.0 上部署的模型導出 API
  

• 用于部署在 NVIDIA Clara 平臺上的醫(yī)學成像應用的模型導出 API
  

讓基于深度學習工作流的智能視頻分析變得可能

對于設計和集成智能視頻分析（IVA）端應用程序（如停車管理、安全基礎(chǔ)設施、零售分析、物流管理和訪問控制等）的開發(fā)人員，NVIDIA 的遷移學習工具包提供了端到端的深度學習工作流，可以加速深度學習訓練，并在 TeslaGPU 上使用 DeepStream SDK 3.0 進行部署。這些模型針對 IVA 特定參考使用場景（如檢測和分類）進行了全面地訓練。

用于 IVA 的遷移學習工具包中已經(jīng)包括了下面這些預訓練的圖像分類與目標檢測模型：

醫(yī)學影像的端到端深度學習流程

醫(yī)學影像的遷移學習工具包提供醫(yī)學影像特有的預訓練模型，以及諸如與 AI 輔助注釋 SDK 集成以加速醫(yī)學圖像的標注的額外功能，從而允許開發(fā)人員使用 AI 輔助標記功能。NVIDIA 研究人員開發(fā)的三維腦腫瘤分割模型贏得了 2018 年多模態(tài)腦腫瘤分割挑戰(zhàn)賽的第一名。BraTS 專注于評估在多模態(tài) MRI 掃描中用于分割腦腫瘤的最新方法。BraTS 2018 利用多機構(gòu)術(shù)前 MRI 掃描，并著重于對內(nèi)在異質(zhì)性（外觀、形狀和組織學）腦腫瘤的分割。NVIDIA 醫(yī)學成像遷移學習工具包附帶了這個由 NVIDIA 內(nèi)部研究人員開發(fā)的獲獎的腦腫瘤分割模型，以及用于肝臟病變分割、脾臟分割等其他預訓練模型。NVIDIA 的醫(yī)學成像端到端深度學習工作流允許開發(fā)人員使用遷移學習工具包來加速深度學習訓練并使用 Clara 平臺進行部署。

雷鋒網(wǎng) AI 科技評論編譯。

雷峰網(wǎng)版權(quán)文章，未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。