TorchCV基于 PyTorch 的計算機視覺深度學習框架

TorchCV 是一個基于 PyTorch 的計算機視覺深度學習框架。該庫提供了基于深度學習的大部分 CV 問題研究的源代碼，對于使用者來說，調用最常用、最為先進的計算機模型從此可以變得更加容易。

TorchCV 支持圖像分類、語義分割、目標檢測、姿態(tài)檢測、實例分割、生成對抗網絡等任務中的多個常見模型，列表如下：

圖像分類：

• 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

 姿態(tài)檢測：

• CPM
• OpenPose

實例分割

• Mask R-CNN

生成對抗網絡

• Pix2pix
• CycleGAN