TorchCV 是一个基于PyTorch的计算机视觉深度学习框架。该库提供了基于深度学习的大部分CV问题研究的源代码,对于使用者来说,调用最常用、最为先进的计算机模型从此可以变得更加容易。
TorchCV支持图像分类、语义分割、目标检测、姿态检测、实例分割、生成对抗网络等任务中的多个常见模型,列表如下:
图像分类:
VGGResNetDenseNetShuffleNetShuffleNetV2PartialOrderPruning语义分割:
DeepLabV3PSPNetDenseASPPAsymmetricNon-localNeuralNetworksforSemanticSegmentation目标检测:
SSDFasterR-CNNYOLOv3FPN姿态检测:
CPMOpenPose实例分割
MaskR-CNN生成对抗网络
Pix2pixCycleGAN
评论