登陆

章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCV

admin 2019-11-09 306人围观 ,发现0个评论

机器之心收拾

参加:张倩、泽南

在机器学习带来的一切颠覆性技能中,核算机视觉范畴招引了业内人士和学术界最大的重视。

在机器学习带来的一切颠覆性技能中,核算机视觉范畴招引了业内人士和学术界最大的重视。

刚刚推出 1.3 正式版的 PyTorch 风头正劲,人们现已环绕这一深度学习结构开发出了越来越多的东西。最近,一个名为 TorchCV 的核算机视觉模型结构站上了 GitHub 趋势榜。

项目链接:https://github.com/donny章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCVyou/torchcv

该库供给了根据深度学习的大部分 CV 问题研讨的源代码,关于运用者来说,调用最常用、最为先进的核算机模型从此能够变得愈加简单。TorchCV 的作者 Donny You 来自北京大学,是机器感知与智能教育部要点实验室的一名研三学生。

章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCV

TorchCV 支撑的模型

TorchCV 支撑图画分类、语义切割、方针检测、姿势检测、实例切割、生成对立网络等使命中的多个常见模型,列表如下:

图画分类

  • VGG: Very Deep Convolutional Networks for Large-Scale Image Recognition
  • ResNet: Deep Residual Learning for Image Recognition
  • DenseNet: Densely Connected Convolutional Networks
  • ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices
  • ShuffleNet V2: Practical Guidelines for Ecient CNN Architecture Design
  • Partial Order Pruning: for Best Speed/Accuracy Trade-off in Neural Architecture Search

语义切割

  • DeepLabV3: Rethinking Atrous Convolution for Semantic Image Segmentation
  • PSPNet: Pyramid Scene Parsing Network
  • DenseASPP: DenseASPP for Semantic Segmentation in Street Scenes
  • Asymmetric Non-local Neural Networks for Semantic Segmentation

方针检测

  • SSD: Single Shot MultiBox Detector
  • Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
  • YOLOv3: An Increme章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCVntal Improvement
  • FPN: Feature Pyramid Networks for Object Detection

姿势检测

  • CPM: Convolutional Pose Machines
  • OpenPose: Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields

实例切割

  • Mask R-CNN

生成对立网络

  • Pix2pix: Image-to-Image Translation with Conditional Adversarial Nets
  • 旧爱难寻
  • CycleGAN: Unpaired Image-to-Image Translation using 章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCVCycle-Consistent

快速发动 TorchCV

TorchCV 现在仅支撑 Python3.x 和 pytorch 1.0。

pip3install-rrequirements.txtcdextensions

shmake.sh

运用 TorchCV 得到的模型功能数据

以下所示的功能数据彻底复现了论文的成果。

图画分类

语义切割

方针检测

姿势估量

  • OpenPose: Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields

实例切割

  • Mask R-CNN

生成对立网

  • Pix2pix
  • CycleGAN

TorchCV 的数据集

TorchCV 界说了你能在数据集子目录中查到的一切使命的数据格式。下面是一个用于练习语义切割的数据集目录树示例。你能够运用 datasets/seg/preprocess文件夹中的脚本对开源数据集进行预处章鱼彩票网-根据PyTorch的CV模型结构,北大学生出品TorchCV理。

文件夹地址:https://github.com/donnyyou/torchcv/tree/master/datasets/seg/preprocess

TorchCV 的指令

项目作者供给了指令行东西,供用户运用模型,代码如下:

练习

cds/seg/cityscapes/

bash run_fs_pspnet_cityscapes_seg.sh train tag

康复练习

cds/seg/cityscapes/

bash run_fs_pspnet_cityscapes_seg.sh train tag

验证

cds/seg/cityscapes/

bash run_fs_pspnet_cityscapes_seg.sh val tag

测验

cds/seg/cityscapes/

bash run_fs_pspnet_cityscapes_seg.sh testtag

2019 年,PyTorch 正在逐步赶超 TensorFlow,在研讨范畴成为最为盛行的深度学习结构。跟着后者运用人数的增多,咱们未来还会看到更多高效东西。

10月16日晚,NVIDIA GPU 核算专家团队高级工程师季光博士将带来线上主题共享:运用 TensorRT 自在建立高功能推理模型。点击阅览原文当即免费报名。

请关注微信公众号
微信二维码
不容错过
Powered By Z-BlogPHP