TorchCV基于 PyTorch 的计算机视觉深度学习框架

TorchCV 是一个基于 PyTorch 的计算机视觉深度学习框架。该库提供了基于深度学习的大部分 CV 问题研究的源代码，对于使用者来说，调用最常用、最为先进的计算机模型从此可以变得更加容易。

TorchCV 支持图像分类、语义分割、目标检测、姿态检测、实例分割、生成对抗网络等任务中的多个常见模型，列表如下：

图像分类：

• VGG
• ResNet
• DenseNet
• ShuffleNet
• ShuffleNet V2
• Partial Order Pruning

语义分割：

• DeepLabV3
• PSPNet
• DenseASPP
• Asymmetric Non-local Neural Networks for Semantic Segmentation

目标检测：

• SSD
• Faster R-CNN
• YOLOv3
• FPN

 姿态检测：

• CPM
• OpenPose

实例分割

• Mask R-CNN

生成对抗网络

• Pix2pix
• CycleGAN