torchvision 接口支持列表
torchvision.transforms
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.transforms.Compose | 支持 | |
| torchvision.transforms.ToTensor | 支持 | 需配合Compose接口使用，否则返回ndarray类型； |
| torchvision.transforms.PILToTensor | 支持 | 需配合Compose接口使用，否则返回ndarray类型； |
| torchvision.transforms.ConvertImageDtype | 支持 | |
| torchvision.transforms.ToPILImage | 支持 | |
| torchvision.transforms.Normalize | 支持 | |
| torchvision.transforms.Resize | 支持 | 暂不支持配置interpolation为BICUBIC；暂不支持配置antialias；暂不支持float64类型输入； |
| torchvision.transforms.CenterCrop | 支持 | |
| torchvision.transforms.Pad | 支持 | |
| torchvision.transforms.Lambda | 支持 | |
| torchvision.transforms.RandomApply | 支持 | |
| torchvision.transforms.RandomChoice | 支持 | |
| torchvision.transforms.RandomOrder | 支持 | |
| torchvision.transforms.RandomCrop | 支持 | 暂不支持配置interpolation为BICUBIC； |
| torchvision.transforms.RandomHorizontalFlip | 支持 | |
| torchvision.transforms.RandomVerticalFlip | 支持 | |
| torchvision.transforms.RandomResizedCrop | 支持 | |
| torchvision.transforms.FiveCrop | 支持 | |
| torchvision.transforms.TenCrop | 支持 | |
| torchvision.transforms.LinearTransformation | 支持 | |
| torchvision.transforms.ColorJitter | 支持 | |
| torchvision.transforms.RandomRotation | 支持 | |
| torchvision.transforms.RandomAffine | 支持 | |
| torchvision.transforms.Grayscale | 支持 | |
| torchvision.transforms.RandomGrayscale | 支持 | |
| torchvision.transforms.RandomPerspective | 支持 | |
| torchvision.transforms.RandomErasing | 支持 | |
| torchvision.transforms.GaussianBlur | 支持 | |
| torchvision.transforms.InterpolationMode | 支持 | |
| torchvision.transforms.RandomInvert | 支持 | |
| torchvision.transforms.RandomPosterize | 支持 | |
| torchvision.transforms.RandomSolarize | 支持 | |
| torchvision.transforms.RandomAdjustSharpness | 支持 | |
| torchvision.transforms.RandomAutocontrast | 支持 | |
| torchvision.transforms.RandomEqualize | 支持 | |
| torchvision.transforms.AutoAugmentPolicy | 支持 | |
| torchvision.transforms.AutoAugment | 支持 | |
| torchvision.transforms.RandAugment | 支持 | |
| torchvision.transforms.TrivialAugmentWide | 支持 | |
| torchvision.transforms.AugMix | 不支持 |
请注意，以上Transforms相关接口只适用于数据预处理阶段，不推荐用于模型训练阶段。
torchvision.io
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.io.write_video | 支持 | |
| torchvision.io.read_video | 支持 | |
| torchvision.io.ImageReadMode | 支持 | |
| torchvision.io.decode_image | 支持 | |
| torchvision.io.decode_jpeg | 支持 | |
| torchvision.io.decode_png | 支持 | |
| torchvision.io.encode_jpeg | 支持 | |
| torchvision.io.encode_png | 支持 | |
| torchvision.io.read_file | 支持 | |
| torchvision.io.read_file | 支持 | |
| torchvision.io.read_file | 支持 | |
| torchvision.io.read_file | 支持 | |
| torchvision.io.write_file | 支持 | |
| torchvision.io.write_jpeg | 支持 | |
| torchvision.io.write_png | 支持 | |
| torchvision.io.Video | 不支持 | |
| torchvision.io.VideoReader | 不支持 | |
| torchvision.io._read_video_clip_from_memory | 不支持 | |
| torchvision.io._read_video_meta_data | 不支持 | |
| torchvision.io.VideoMetaData | 不支持 | |
| torchvision.io.Timebase | 不支持 | |
| torchvision.io._read_video_from_file | 不支持 | |
| torchvision.io._read_video_timestamps_from_file | 不支持 | |
| torchvision.io._probe_video_from_file | 不支持 | |
| torchvision.io._read_video_from_memory | 不支持 | |
| torchvision.io._read_video_timestamps_from_memory | 不支持 | |
| torchvision.io._probe_video_from_memory | 不支持 | |
| torchvision.io._HAS_VIDEO_OPT | 不支持 |
torchvision.datasets
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.datasets.LSUN | 支持 | |
| torchvision.datasets.LSUNClass | 支持 | |
| torchvision.datasets.ImageFolder | 支持 | |
| torchvision.datasets.DatasetFolder | 支持 | |
| torchvision.datasets.FakeData | 支持 | |
| torchvision.datasets.CocoCaptions | 支持 | |
| torchvision.datasets.CocoDetection | 支持 | |
| torchvision.datasets.CIFAR10 | 支持 | |
| torchvision.datasets.CIFAR100 | 支持 | |
| torchvision.datasets.EMNIST | 支持 | |
| torchvision.datasets.FashionMNIST | 支持 | |
| torchvision.datasets.QMNIST | 支持 | |
| torchvision.datasets.MNIST | 支持 | |
| torchvision.datasets.KMNIST | 支持 | |
| torchvision.datasets.StanfordCars | 支持 | |
| torchvision.datasets.STL10 | 支持 | |
| torchvision.datasets.SUN397 | 支持 | |
| torchvision.datasets.SVHN | 支持 | |
| torchvision.datasets.PhotoTour | 支持 | |
| torchvision.datasets.SEMEION | 支持 | |
| torchvision.datasets.Omniglot | 支持 | |
| torchvision.datasets.LSUN | 支持 | |
| torchvision.datasets.LSUNClass | 支持 | |
| torchvision.datasets.ImageFolder | 支持 | |
| torchvision.datasets.DatasetFolder | 支持 | |
| torchvision.datasets.FakeData | 支持 | |
| torchvision.datasets.CocoCaptions | 支持 | |
| torchvision.datasets.CocoDetection | 支持 | |
| torchvision.datasets.CIFAR10 | 支持 | |
| torchvision.datasets.CIFAR100 | 支持 | |
| torchvision.datasets.EMNIST | 支持 | |
| torchvision.datasets.FashionMNIST | 支持 | |
| torchvision.datasets.QMNIST | 支持 | |
| torchvision.datasets.MNIST | 支持 | |
| torchvision.datasets.KMNIST | 支持 | |
| torchvision.datasets.StanfordCars | 支持 | |
| torchvision.datasets.STL10 | 支持 | |
| torchvision.datasets.SUN397 | 支持 | |
| torchvision.datasets.SVHN | 支持 | |
| torchvision.datasets.PhotoTour | 支持 | |
| torchvision.datasets.SEMEION | 支持 | |
| torchvision.datasets.Omniglot | 支持 | |
| torchvision.datasets.SBU | 支持 | |
| torchvision.datasets.Flickr8k | 支持 | |
| torchvision.datasets.Flickr30k | 支持 | |
| torchvision.datasets.Flowers102 | 支持 | |
| torchvision.datasets.VOCSegmentation | 支持 | |
| torchvision.datasets.VOCDetection | 支持 | |
| torchvision.datasets.Cityscapes | 支持 | |
| torchvision.datasets.ImageNet | 支持 | |
| torchvision.datasets.Caltech101 | 支持 | 内置下载链接在googledrive，如果不能自动下载请按照提示官方地址下载； |
| torchvision.datasets.Caltech256 | 支持 | 内置下载链接在googledrive，如果不能自动下载请按照提示官方地址下载； |
| torchvision.datasets.CelebA | 支持 | 内置下载链接在googledrive，如果不能自动下载请按照提示官方地址下载； |
| torchvision.datasets.WIDERFace | 支持 | |
| torchvision.datasets.SBDataset | 支持 | |
| torchvision.datasets.VisionDataset | 支持 | |
| torchvision.datasets.USPS | 支持 | |
| torchvision.datasets.Kinetics400 | 支持 | |
| torchvision.datasets.Kinetics | 支持 | |
| torchvision.datasets.HMDB51 | 支持 | |
| torchvision.datasets.UCF101 | 支持 | |
| torchvision.datasets.Places365 | 支持 | |
| torchvision.datasets.Kitti | 支持 | |
| torchvision.datasets.INaturalist | 支持 | |
| torchvision.datasets.LFWPeople | 支持 | |
| torchvision.datasets.LFWPairs | 支持 | |
| torchvision.datasets.KittiFlow | 支持 | |
| torchvision.datasets.Sintel | 支持 | |
| torchvision.datasets.FlyingChairs | 支持 | |
| torchvision.datasets.FlyingThings3D | 支持 | |
| torchvision.datasets.HD1K | 支持 | |
| torchvision.datasets.Food101 | 支持 | |
| torchvision.datasets.DTD | 支持 | |
| torchvision.datasets.FER2013 | 支持 | |
| torchvision.datasets.GTSRB | 支持 | |
| torchvision.datasets.CLEVRClassification | 支持 | |
| torchvision.datasets.OxfordIIITPet | 支持 | |
| torchvision.datasets.PCAM | 支持 | 内置下载链接在googledrive，如果不能自动下载请按照提示官方地址下载； |
| torchvision.datasets.Country211 | 支持 | |
| torchvision.datasets.FGVCAircraft | 支持 | |
| torchvision.datasets.EuroSAT | 支持 | |
| torchvision.datasets.RenderedSST2 | 支持 | |
| torchvision.datasets.SBU | 支持 | |
| torchvision.datasets.Flickr8k | 支持 | |
| torchvision.datasets.Flickr30k | 支持 | |
| torchvision.datasets.Flowers102 | 支持 | |
| torchvision.datasets.VOCSegmentation | 支持 | |
| torchvision.datasets.VOCDetection | 支持 | |
| torchvision.datasets.Cityscapes | 支持 | |
| torchvision.datasets.ImageNet | 支持 | |
| torchvision.datasets.Caltech101 | 支持 | |
| torchvision.datasets.Caltech256 | 支持 | |
| torchvision.datasets.CelebA | 支持 | |
| torchvision.datasets.WIDERFace | 支持 | |
| torchvision.datasets.SBDataset | 支持 | |
| torchvision.datasets.VisionDataset | 支持 | |
| torchvision.datasets.USPS | 支持 | |
| torchvision.datasets.Kinetics400 | 支持 | |
| torchvision.datasets.Kinetics | 支持 | |
| torchvision.datasets.HMDB51 | 支持 | |
| torchvision.datasets.UCF101 | 支持 | |
| torchvision.datasets.Places365 | 支持 | |
| torchvision.datasets.Kitti | 支持 | |
| torchvision.datasets.INaturalist | 支持 | |
| torchvision.datasets.LFWPeople | 支持 | |
| torchvision.datasets.LFWPairs | 支持 | |
| torchvision.datasets.KittiFlow | 支持 | |
| torchvision.datasets.Sintel | 支持 | |
| torchvision.datasets.FlyingChairs | 支持 | |
| torchvision.datasets.FlyingThings3D | 支持 | |
| torchvision.datasets.HD1K | 支持 | |
| torchvision.datasets.Food101 | 支持 | |
| torchvision.datasets.DTD | 支持 | |
| torchvision.datasets.FER2013 | 支持 | |
| torchvision.datasets.GTSRB | 支持 | |
| torchvision.datasets.CLEVRClassification | 支持 | |
| torchvision.datasets.OxfordIIITPet | 支持 | |
| torchvision.datasets.PCAM | 支持 | |
| torchvision.datasets.Country211 | 支持 | |
| torchvision.datasets.FGVCAircraft | 支持 | |
| torchvision.datasets.EuroSAT | 支持 | |
| torchvision.datasets.RenderedSST2 | 支持 |
torchvision.models
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.models.AlexNet | 支持 | |
| torchvision.models.alexnet | 支持 | |
| torchvision.models.DenseNet | 支持 | |
| torchvision.models.densenet121 | 支持 | |
| torchvision.models.densenet169 | 支持 | |
| torchvision.models.densenet201 | 支持 | |
| torchvision.models.densenet161 | 支持 | |
| torchvision.models.GoogLeNet | 支持 | |
| torchvision.models.googlenet | 支持 | |
| torchvision.models.Inception3 | 支持 | |
| torchvision.models.inception_v3 | 支持 | |
| torchvision.models.MNASNet | 支持 | |
| torchvision.models.mnasnet0_5 | 支持 | |
| torchvision.models.mnasnet0_75 | 支持 | |
| torchvision.models.mnasnet1_0 | 支持 | |
| torchvision.models.mnasnet1_3 | 支持 | |
| torchvision.models.MobileNetV2 | 支持 | |
| torchvision.models.mobilenet_v2 | 支持 | |
| torchvision.models.MobileNetV3 | 支持 | |
| torchvision.models.mobilenet_v3_large | 支持 | |
| torchvision.models.mobilenet_v3_small | 支持 | |
| torchvision.models.ResNet | 支持 | |
| torchvision.models.resnet18 | 支持 | |
| torchvision.models.resnet34 | 支持 | |
| torchvision.models.resnet50 | 支持 | |
| torchvision.models.resnet101 | 支持 | |
| torchvision.models.resnet152 | 支持 | |
| torchvision.models.resnext50_32x4d | 支持 | |
| torchvision.models.resnext101_32x8d | 支持 | |
| torchvision.models.wide_resnet50_2 | 支持 | |
| torchvision.models.wide_resnet101_2 | 支持 | |
| torchvision.models.ShuffleNetV2 | 支持 | |
| torchvision.models.shufflenet_v2_x0_5 | 支持 | |
| torchvision.models.shufflenet_v2_x1_0 | 支持 | |
| torchvision.models.shufflenet_v2_x1_5 | 支持 | |
| torchvision.models.shufflenet_v2_x2_0 | 支持 | |
| torchvision.models.SqueezeNet | 支持 | |
| torchvision.models.squeezenet1_0 | 支持 | |
| torchvision.models.squeezenet1_1 | 支持 | |
| torchvision.models.VGG | 支持 | |
| torchvision.models.vgg11 | 支持 | |
| torchvision.models.vgg11_bn | 支持 | |
| torchvision.models.vgg13 | 支持 | |
| torchvision.models.vgg13_bn | 支持 | |
| torchvision.models.vgg16 | 支持 | |
| torchvision.models.vgg16_bn | 支持 | |
| torchvision.models.vgg19_bn | 支持 | |
| torchvision.models.vgg19 | 支持 | |
| torchvision.models.detection.FasterRCNN | 支持 | |
| torchvision.models.detection.fasterrcnn_resnet50_fpn | 支持 | |
| torchvision.models.detection.fasterrcnn_mobilenet_v3_large_320_fpn | 支持 | |
| torchvision.models.detection.fasterrcnn_mobilenet_v3_large_fpn | 支持 | |
| torchvision.models.detection.KeypointRCNN | 支持 | |
| torchvision.models.detection.keypointrcnn_resnet50_fpn | 支持 | |
| torchvision.models.detection.MaskRCNN | 支持 | |
| torchvision.models.detection.maskrcnn_resnet50_fpn | 支持 | |
| torchvision.models.detection.RetinaNet | 支持 | |
| torchvision.models.detection.retinanet_resnet50_fpn | 支持 | |
| torchvision.models.detection.SSD | 支持 | |
| torchvision.models.detection.ssd300_vgg16 | 支持 | |
| torchvision.models.detection.ssdlite320_mobilenet_v3_large | 支持 | |
| torchvision.models.segmentation.DeepLabV3 | 支持 | |
| torchvision.models.segmentation.FCN | 支持 | |
| torchvision.models.segmentation.LRASPP | 支持 | |
| torchvision.models.segmentation.fcn_resnet50 | 支持 | |
| torchvision.models.segmentation.fcn_resnet101 | 支持 | |
| torchvision.models.segmentation.deeplabv3_resnet50 | 支持 | |
| torchvision.models.segmentation.deeplabv3_resnet101 | 支持 | |
| torchvision.models.segmentation.deeplabv3_mobilenet_v3_large | 支持 | |
| torchvision.models.segmentation.lraspp_mobilenet_v3_large | 支持 | |
| torchvision.models.video.r3d_18 | 支持 | |
| torchvision.models.video.mc3_18 | 支持 | |
| torchvision.models.video.r2plus1d_18 | 支持 | |
| torchvision.models.convnext.ConvNeXt | 支持 | |
| torchvision.models.convnext.convnext_tiny | 支持 | |
| torchvision.models.convnext.convnext_small | 支持 | |
| torchvision.models.convnext.convnext_base | 支持 | |
| torchvision.models.convnext.convnext_large | 支持 | |
| mindtorch.torchvision.models.efficientnet.EfficientNet | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b0 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b1 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b2 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b3 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b4 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b5 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b6 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_b7 | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_v2_s | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_v2_m | 支持 | |
| mindtorch.torchvision.models.efficientnet.efficientnet_v2_l | 支持 | |
| torchvision.models.regnet.RegNet | 支持 | |
| torchvision.models.regnet.regnet_y_400mf | 支持 | |
| torchvision.models.regnet.regnet_y_800mf | 支持 | |
| torchvision.models.regnet.regnet_y_1_6gf | 支持 | |
| torchvision.models.regnet.regnet_y_3_2gf | 支持 | |
| torchvision.models.regnet.regnet_y_8gf | 支持 | |
| torchvision.models.regnet.regnet_y_16gf | 支持 | |
| torchvision.models.regnet.regnet_y_32gf | 支持 | |
| torchvision.models.regnet.regnet_y_128gf | 支持 | |
| torchvision.models.regnet.regnet_x_400mf | 支持 | |
| torchvision.models.regnet.regnet_x_800mf | 支持 | |
| torchvision.models.regnet.regnet_x_1_6gf | 支持 | |
| torchvision.models.regnet.regnet_x_3_2gf | 支持 | |
| torchvision.models.regnet.regnet_x_8gf | 支持 | |
| torchvision.models.regnet.regnet_x_16gf | 支持 | |
| torchvision.models.regnet.regnet_x_32gf | 支持 | |
| torchvision.models.swin_transformer.SwinTransformer | 支持 | |
| torchvision.models.swin_transformer.swin_t | 支持 | |
| torchvision.models.swin_transformer.swin_s | 支持 | |
| torchvision.models.swin_transformer.swin_b | 支持 | |
| mindtorch.torchvision.models.vision_transformer.VisionTransformer | 支持 | |
| mindtorch.torchvision.models.vision_transformer.vit_b_16 | 支持 | |
| mindtorch.torchvision.models.vision_transformer.vit_b_32 | 支持 | |
| mindtorch.torchvision.models.vision_transformer.vit_l_16 | 支持 | |
| mindtorch.torchvision.models.vision_transformer.vit_l_32 | 支持 | |
| mindtorch.torchvision.models.vision_transformer.vit_h_14 | 支持 | |
| mindtorch.torchvision.models.detection.FCOS | 支持 | |
| mindtorch.torchvision.models.detection.fcos_resnet50_fpn | 支持 | |
| mindtorch.torchvision.models.optical_flow.RAFT | 支持 | |
| mindtorch.torchvision.models.optical_flow.raft_large | 支持 | |
| mindtorch.torchvision.models.optical_flow.raft_small | 支持 | |
| torchvision.models.quantization.QuantizableGoogLeNet | 不支持 | |
| torchvision.models.quantization.googlenet | 不支持 | |
| torchvision.models.quantization.QuantizableInception3 | 不支持 | |
| torchvision.models.quantization.inception_v3 | 不支持 | |
| torchvision.models.quantization.QuantizableMobileNetV2 | 不支持 | |
| torchvision.models.quantization.mobilenet_v2 | 不支持 | |
| torchvision.models.quantization.QuantizableMobileNetV3 | 不支持 | |
| torchvision.models.quantization.mobilenet_v3_large | 不支持 | |
| torchvision.models.quantization.QuantizableResNet | 不支持 | |
| torchvision.models.quantization.resnet18 | 不支持 | |
| torchvision.models.quantization.resnet50 | 不支持 | |
| torchvision.models.quantization.resnext101_32x8d | 不支持 | |
| torchvision.models.quantization.QuantizableShuffleNetV2 | 不支持 | |
| torchvision.models.quantization.shufflenet_v2_x0_5 | 不支持 | |
| torchvision.models.quantization.shufflenet_v2_x1_0 | 不支持 | |
| torchvision.models.quantization.shufflenet_v2_x1_5 | 不支持 | |
| torchvision.models.quantization.shufflenet_v2_x2_0 | 不支持 | |
| torchvision.models.feature_extraction.create_feature_extractor | 不支持 | |
| torchvision.models.feature_extraction.get_graph_node_names | 不支持 |
torchvision.ops
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.ops.masks_to_boxes | 支持 | |
| torchvision.ops.deform_conv2d | 支持 | |
| torchvision.ops.DeformConv2d | 支持 | |
| torchvision.ops.nms | 支持 | |
| torchvision.ops.batched_nms | 支持 | |
| torchvision.ops.remove_small_boxes | 支持 | |
| torchvision.ops.clip_boxes_to_image | 支持 | |
| torchvision.ops.box_convert | 支持 | |
| torchvision.ops.box_area | 支持 | |
| torchvision.ops.box_iou | 支持 | |
| torchvision.ops.generalized_box_iou | 支持 | |
| torchvision.ops.distance_box_iou | 支持 | |
| torchvision.ops.complete_box_iou | 支持 | |
| torchvision.ops.roi_align | 支持 | 当前sampling_ratio取值需大于0；暂不支持配置 aligned； |
| torchvision.ops.RoIAlign | 支持 | 当前sampling_ratio取值需大于0；暂不支持配置 aligned； |
| torchvision.ops.MultiScaleRoIAlign | 支持 | |
| torchvision.ops.FeaturePyramidNetwork | 支持 | |
| torchvision.ops.sigmoid_focal_loss | 支持 | |
| torchvision.ops.stochastic_depth | 支持 | |
| torchvision.ops.StochasticDepth | 支持 | |
| torchvision.ops.FrozenBatchNorm2d | 支持 | |
| torchvision.ops.Conv2dNormActivation | 支持 | |
| torchvision.ops.Conv3dNormActivation | 支持 | |
| torchvision.ops.SqueezeExcitation | 支持 | |
| torchvision.ops.MLP | 支持 | |
| torchvision.ops.Permute | 支持 | |
| torchvision.ops.generalized_box_iou_loss | 支持 | |
| torchvision.ops.distance_box_iou_loss | 支持 | |
| torchvision.ops.complete_box_iou_loss | 支持 | |
| torchvision.ops.drop_block2d | 支持 | |
| torchvision.ops.DropBlock2d | 支持 | |
| torchvision.ops.drop_block3d | 支持 | |
| torchvision.ops.DropBlock3d | 支持 | |
| torchvision.ops.roi_pool | 不支持 | |
| torchvision.ops.RoIPool | 不支持 | |
| torchvision.ops.ps_roi_align | 不支持 | |
| torchvision.ops.PSRoIAlign | 不支持 | |
| torchvision.ops.ps_roi_pool | 不支持 | |
| torchvision.ops.PSRoIPool | 不支持 |
torchvision.utils
| MindTorch Torchvision 接口 | 状态 | 备注 |
|---|---|---|
| torchvision.utils.make_grid | 支持 | |
| torchvision.utils.save_image | 支持 | |
| torchvision.utils.draw_bounding_boxes | 支持 | |
| torchvision.utils.draw_segmentation_masks | 支持 | |
| torchvision.utils.draw_keypoints | 支持 | |
| torchvision.utils.flow_to_image | 支持 |