实现基于pytorch的数据增强的代码

最新推荐文章于 2023-12-05 11:44:54 发布

耄先森吖

最新推荐文章于 2023-12-05 11:44:54 发布

阅读量152

点赞数

文章标签： pytorch 深度学习人工智能 python 计算机视觉

本文链接：https://blog.csdn.net/weixin_35753431/article/details/129489310

版权

以下是一个基于PyTorch的数据增强代码示例，使用了torchvision.transforms模块中的一些预定义的转换函数。

import torchvision.transforms as transforms

# 定义一个数据增强的转换函数
transform = transforms.Compose([
    transforms.RandomHorizontalFlip(), # 随机水平翻转
    transforms.RandomRotation(10),     # 随机旋转10度以内
    transforms.ColorJitter(brig