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}
}