Advertisement

神经网络转换数据预处理

阅读量:

torchvision 中 transforms 的使用

常规编程方式的编写

复制代码
    from torchvision import transforms
    
    data_transform = {"train": transforms.Compose([
    									 transforms.RandomResizedCrop(224), 
                                     transforms.RandomHorizontalFlip(), 
                                     transforms.ToTensor(),
                                     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])}
    
    
    AI写代码python
    
    运行

transforms.RandomResizedCrop()

复制代码
    """
    A crop of random size (default: of 0.08 to 1.0) of the original size and a rando

全部评论 (0)

还没有任何评论哟~