PyTorch开源物体检测工具包

MMDetection是一个基于PyTorch的开源物体检测工具包。该工具包采用模块化设计，支持多种流行的物体检测和实例分割算法，并且可以灵活地进行拓展，在速度和显存消耗上也具有优势。(https://github.com/open-mmlab/mmdetection)

目前已经支持单阶段检测器如SSD/RetinaNet/FCOS/FSAF，两阶段检测器如FasterR-CNN/Mask R-CNN，多阶段检测器如Cascade R-CNN/Hybrid Task Cascade等，另外支持许多相关模块如DCN/Soft-NMS/OHEM等，也支持混合精度训练。有很多最新的工作也在MMDetection上开源。

团队为之提供了完整的训练和测试框架，以及超过200个训练好的模型及其测试结果，希望能为社区提供统一的开发平台和测试基准，助力物体检测的相关研究。

MMAction是一个基于Pytorch的开源视频动作理解工具包，囊括了视频动作分类、时域动作检测（定位）、时空动作检测等视频理解的基础任务。(https://github.com/open-mmlab/mmaction)

目前已经支持双流、TSN、SSN等动作分类和动作检测框架和基于Fast R-CNN的时空动作检测基线模型，支持Plain 2D/Inflated3D/Non-local等流行的网络结构，支持UCF-101、Something-Something、Kinetics、THUMOS14、ActivityNet、AVA等视频数据集，并提供相关的预训练模型。

作为Open-MMLab系统开源项目的一部分，团队希望MMAction可以成为视频研究人员的测试平台，促进视频动作理解领域更上新台阶。

PyTorch开源物体检测工具包相关推荐

港中大、商汤开源目标检测工具包mmdetection，对比Detectron如何？
近日,香港中文大学-商汤联合实验室开源了基于 PyTorch 的检测库--mmdetection.上个月,商汤和港中大组成的团队在 COCO 比赛的物体检测(Detection)项目中夺得冠军,而 m ...
pytorch模型加载测试_使用Pytorch实现物体检测（Faster R-CNN）
在本示例中,介绍一种two-stage算法(Faster R-CNN),将目标区域检测和类别识别分为两个任务进行物体检测.本示例采用PyTorch引擎进行模型构建. 如果您已熟练使用Notebook和 ...
[pytorch] RCNN物体检测的简化实现（d2l-香蕉数据集）
论文地址 RCNN物体检测的主要步骤如下: 训练: 监督预训练:利用ILSVRC 2012进行预训练,lr=0.01 在VOC 2012 train上微调:将CNN最后的1000分类初始化为21分类( ...
BBAug: PyTorch的物体检测包
点上方计算机视觉联盟获取更多干货仅作学术分享,不代表本公众号立场,侵权联系删除转载于:AI公园作者:Harpal Sahota 编译:ronghuaiyang AI博士笔记系列推荐周志 ...
BBAug: 一个用于PyTorch的物体检测包围框数据增强包
本文转载自AI公园. 作者:Harpal Sahota 编译:ronghuaiyang 导读实现了Google Research,Brain Team中的增强策略. 像许多神经网络模型一样,目标检测 ...
ros 开源物体检测_ROS传感器之LIDAR简介
一.概念 LIDAR,全称Light Detection And Ranging,即激光检测和测距,通常称为激光雷达.其功能是利用激光检测物体以及测量与物体之间的距离. 二.分类不同的分类标准,有不 ...
深度学习二(Pytorch物体检测实战)
深度学习二(Pytorch物体检测实战) 文章目录深度学习二(Pytorch物体检测实战) 1.PyTorch基础 1.1.基本数据结构:Tensor 1.1.1.Tensor数据类型 1.1.2. ...
二十. 在ROS系统上实现基于PyTorch YOLO v5的实时物体检测
一. 背景介绍在我前面的博文十八.在JetsonNano上为基于PyTorch的物体检测网络测速和选型中,我介绍过在基于Jetson Nano硬件平台和Ubuntu 18.04 with Jet ...
对象检测工具包mmdetection简介、安装及测试代码
mmdetection是商汤和港中文大学联合开源的基于PyTorch的对象检测工具包,属于香港中文大学多媒体实验室open-mmlab项目的一部分.该工具包提供了已公开发表的多种流行的检测组件,通过这 ...

PyTorch开源物体检测工具包

PyTorch开源物体检测工具包相关推荐

最新文章

热门文章