CVNets計(jì)算機(jī)視覺網(wǎng)絡(luò)訓(xùn)練庫

CVNets 是一個計(jì)算機(jī)視覺工具包，它允許研究人員和工程師為各種任務(wù)訓(xùn)練標(biāo)準(zhǔn)和移動/非移動計(jì)算機(jī)視覺模型，包括對象分類、對象檢測、語義分割和基礎(chǔ)模型（例如，CLIP）。

安裝

建議使用 Python 3.8+ 和PyTorch（版本 >= v1.12.0）

以下說明使用 Conda，如果沒有安裝 Conda，可以查看如何安裝 Conda。

# Clone the repo git clone [email protected]:apple/ml-cvnets.git cd ml-cvnets # Create a virtual env. We use Conda conda create -n cvnets python=3.8 conda activate cvnets # install requirements and CVNets package pip install -r requirements.txt pip install --editable .

入門

• 此處給出了使用 CVNet 的一般說明。
• 此處和此處提供了訓(xùn)練和評估模型的示例。
• 此處提供了將 PyTorch 模型轉(zhuǎn)換為 CoreML 的示例。

支持的模型和任務(wù)

要查看可用模型和基準(zhǔn)的列表，請參閱Model Zoo和示例文件夾。