CVNets计算机视觉网络训练库

联合创作 · 2023-09-25 22:57

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 .

入门

  • 此处给出了使用 CVNet 的一般说明。
  • 此处此处提供了训练和评估模型的示例。
  • 此处提供了将 PyTorch 模型转换为 CoreML 的示例。

支持的模型和任务

要查看可用模型和基准的列表,请参阅Model Zoo示例文件夹。

浏览 14
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报