ml cvnets
1.0.0
CVNET是一種計算機視覺工具包,可允許研究人員和工程師培訓標準和新型的移動和非移動計算機視覺模型,用於各種任務,包括對象分類,對象檢測,語義細分和基礎模型(例如,剪輯)。
我們建議使用Python 3.10+和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.10.8
conda activate cvnets
# install requirements and CVNets package
pip install -r requirements.txt -c constraints.txt
pip install --editable . 要查看可用型號和基準的列表,請參閱模型動物園和示例文件夾。
該代碼由Sachin開發,現在由Sachin,Maxwell Horton,Mohammad Sekhavat和Yanzi Jin維護。
以下是使用CVNET的Apple出版物清單:
我們歡迎社區的公關!您可以在我們的貢獻文件中找到有關為CVNET貢獻的信息。
請記住遵循我們的行為準則。
有關許可證詳細信息,請參閱許可證。
如果您發現我們的工作有用,請引用以下論文:
@inproceedings{mehta2022mobilevit,
title={MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer},
author={Sachin Mehta and Mohammad Rastegari},
booktitle={International Conference on Learning Representations},
year={2022}
}
@inproceedings{mehta2022cvnets,
author = {Mehta, Sachin and Abdolhosseini, Farzad and Rastegari, Mohammad},
title = {CVNets: High Performance Library for Computer Vision},
year = {2022},
booktitle = {Proceedings of the 30th ACM International Conference on Multimedia},
series = {MM '22}
}