mmrazor

文檔|安裝| ？模型動物園|報告問題

英語| 簡體中文

現在可以使用大型型號的Mmrazor ！請參考mmrazorlarge

介紹

Mmrazor是用於模型減肥和汽車的模型壓縮工具包，其中包括4種主流技術：

• 神經建築搜索（NAS）
• 修剪
• 知識蒸餾（KD）
• 量化

它是OpenMMLAB項目的一部分。

主要功能：

• 相容性

  由於OpenMMLAB的類似架構設計以及Slimming算法和視覺任務的解耦，因此MMRAZOR可以輕鬆地應用於OpenMMLAB的各種項目。

• 靈活性

  可以以插頭-n-play的方式合併不同的算法，例如，NAS，修剪和KD，以構建更強大的系統。

• 方便

  使用更好的模塊化設計，開發人員可以僅使用幾個代碼，甚至簡單地修改配置文件，實現新的模型壓縮算法。

關於Mmrazor的設計和實施，請參閱教程以獲取更多詳細信息。

最新更新

默認分支現在是main ，分支上的代碼已升級到v1.0.0。舊的master分支代碼現在存在於0.x分支

Mmrazor v1.0.0於2023-4-24發布，1.0.0rc2的重大更新包括：

1. MMRAZOR量化已發布。
2. 添加一種名為GroupFisher的新修剪算法。
3. 用mmrazor支持蒸餾RTMDET。

要了解有關MMRAZOR 1.0中更新的更多信息，請參閱ChangElog以獲取更多詳細信息！

基準和模型動物園

模型動物園中可用結果和模型。

支持的算法：

神經建築搜索

• 飛鏢（ICL'2019）

• detnas（Neurips'2019）

• SPO（ECCV'2020）

修剪

• Autoslim（Neurips'2019）

• l1-norm

• 集團費舍爾

• DMCP

知識蒸餾

• CWD（ICCV'2021）

• WSLD（ICL'2021）

• 阿布羅斯

• BYOT

• dafl

• DFAD

• DKD

• 因子轉移

• fitnets

• KD

• OFD

• RKD

• ZSKT

• fbkd

量化

• Ptq

• 卡特

• LSQ

安裝

mmrazor取決於pytorch，mmcv和mmengine。

請參閱installation.md以獲取更多詳細說明。

入門

請參閱用戶指南以獲取mmrazor的基本用法。也有高級指南：

貢獻

我們感謝為改善Mmrazor的所有貢獻。請參閱contrufing.md以獲取貢獻指南。

致謝

Mmrazor是一個開源項目，由來自各學院和公司的研究人員和工程師貢獻。我們感謝所有實施方法或添加新功能的貢獻者以及提供寶貴反饋的用戶。我們希望工具箱和基準可以通過提供靈活的工具包來重新實現現有方法並開發自己的新模型壓縮方法來為不斷增長的研究社區提供服務。

引用

如果您發現此項目對您的研究有用，請考慮引用：

 @misc { 2021mmrazor ,
    title = { OpenMMLab Model Compression Toolbox and Benchmark } ,
    author = { MMRazor Contributors } ,
    howpublished = { url{https://github.com/open-mmlab/mmrazor} } ,
    year = { 2021 }
}

執照

該項目以Apache 2.0許可發布。

OpenMMLAB的項目

• MMCV：計算機視覺的OpenMMLAB基礎庫。
• MIM：MIM安裝OpenMMLAB軟件包。
• MMClassification：OpenMMLAB圖像分類工具箱和基準測試。
• mmDetection：OpenMMLAB檢測工具箱和基準測試。
• MMDetection3D：OpenMMLAB的一般3D對象檢測的下一代平台。
• mmrotate：OpenMMLAB旋轉對象檢測工具箱和基準測試。
• Myolo：OpenMMLAB YOLO系列工具箱和基準測試。
• MMSementation：OpenMMLAB語義分割工具箱和基準測試。
• MMOCR：OpenMMLAB文本檢測，識別和理解工具箱。
• mmpose：OpenMMLAB姿勢估計工具箱和基準。
• mmhuman3d：OpenMMLAB 3D人類參數模型工具箱和基準測試。
• MMSelfSup：OpenMMLAB自製學習工具箱和基準。
• mmrazor：OpenMMLAB模型壓縮工具箱和基準測試。
• mmfewshot：OpenMMLAB LIGHSHOT LEALLED TOOLBAX和BENCHMARK。
• MMACTION2：OpenMMLAB的下一代動作理解工具箱和基準測試。
• mmmtracking：OpenMMLAB視頻感知工具箱和基準測試。
• MMFlow：OpenMMLAB光流動工具箱和基準測試。
• mediting：OpenMMLAB圖像和視頻編輯工具箱。
• mmgeneration：OpenMMLAB圖像和視頻生成模型工具箱。
• mmdeploy：OpenMMLAB模型部署框架。