Pythorch图像增强
ImageAug的Python项目详细描述
图像
Pythorch图像增强
- 应用不超出图像边界的随机裁剪旋转
- 将RGB转换为YUV颜色空间
- 调整亮度和对比度等
快速入门
这些转换被设计成使用torchvision.transforms.Compose
链接在一起。此外,还有一个功能模块。如果必须构建更复杂的转换管道,函数转换提供了更细粒度的控制。在
安装
pip3 install -r requirements.txt python3 setup.py install
要求:
- 枕头
- 火炬视觉
- numpy公司
示例
^{pr2}$当前特性
- 在原始图像的范围内旋转和裁剪图像,以获得任何给定角度扰动(对于具有旋转噪声的训练样本)
- 使用alpha通道支持在RGB/YUV/YCH颜色空间之间转换图像
- 调整频道的对比度和亮度
- 噪声发生
待办事项
这个项目仍在进行中。在
- 随机轮作作物的均匀分布
- 颜色查找表用于更快的颜色空间转换
- 添加图像、文本、形状和像素化
项目页面
Github:https://github.com/paixi/ImageAug
- 项目
标签: