非官方PyTorch版本的RandAugment重现：pytorch-randaugment
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
pytorch-randaugment是一个非官方但功能强大的PyTorch库，实现了RandAugment数据增强技术，用于提高机器学习模型在图像分类任务中的性能和泛化能力。
PyTorch-Randaugment 是 RandAugment 的非官方 PyTorch 重新实现版本。大部分代码来自其他开源项目。介绍使用 RandAugment 可以在不依赖单独代理任务的情况下，对感兴趣的数据集训练模型。通过仅调整两个超参数（N 和 M），您可以获得与 AutoAugments 性能相竞争的效果。
安装方法：
```
pip install git+https://github.com/ildoonet/pytorch-randaugment
```
用法示例：
```python
from torchvision.transforms import transforms
from RandAugment import RandAugment
transform_train = transforms.Compose([
transforms.RandomCrop(32, padding=4),
# 其他转换步骤...
])
```
全部评论 (0)
还没有任何评论哟~
