EasyCV 一体化计算机视觉工具箱开源项目

我要开发同款
匿名用户2022年09月07日
30阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议Apache-2.0

作品详情

EasyCV是基于PyTorch的一体化计算机视觉工具箱,主要专注于自监督学习、基于Transformer的模型,以及主要的CV任务,包括图像分类、度量学习、目标检测、姿态估计等。

主要特点

SOTASSL算法

EasyCV在基于对比学习的自我监督学习中提供最先进的算法,例如SimCLR、MoCOV2、Swav、DINO以及基于蒙版图像建模的MAE。我们还提供用于ssl模型评估的标准基准测试工具。

视觉变形金刚

EasyCV旨在提供一种简单的方法来使用通过监督学习或自我监督学习训练的现成SOTA变压器模型,例如ViT、SwinTransformer和DETR系列。未来将添加更多模型。此外,我们支持来自timm的所有预训练模型。

功能和可扩展性

除了SSL,EasyCV还支持图像分类、物体检测、度量学习,未来将支持更多领域。虽然涵盖了不同的领域,但EasyCV将框架分解为不同的组件,例如数据集、模型和运行钩子,从而可以轻松添加新组件并将其与现有模块组合。

EasyCV提供简单而全面的推理接口。此外,PAI-EAS支持所有模型,可以轻松部署为在线服务,并支持自动扩展和服务监控。

效率

EasyCV支持多GPU和多工人训练。EasyCV使用DALI加速数据io和预处理过程,并使用TorchAccelerator和fp16加速训练过程。对于推理优化,EasyCV使用jit脚本导出模型,可以通过PAI-Blade进行优化

什么是新的[
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论