CVNets计算机视觉网络训练库
CVNets 是一个计算机视觉工具包,它允许研究人员和工程师为各种任务训练标准和移动/非移动计算机视觉模型,包括对象分类、对象检测、语义分割和基础模型(例如,CLIP)。
安装
建议使用 Python 3.8+ 和PyTorch(版本 >= v1.12.0)
以下说明使用 Conda,如果没有安装 Conda,可以查看如何安装 Conda。
# Clone the repo git clone git@github.com: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 .
入门
支持的模型和任务
评论