Pythorch图像增强

ImageAug的Python项目详细描述


图像

Pythorch图像增强

  • 应用不超出图像边界的随机裁剪旋转
  • 将RGB转换为YUV颜色空间
  • 调整亮度和对比度等

Example

Artwork由@hcnone

快速入门

这些转换被设计成使用torchvision.transforms.Compose链接在一起。此外,还有一个功能模块。如果必须构建更复杂的转换管道,函数转换提供了更细粒度的控制。在

安装

pip3 install -r requirements.txt
python3 setup.py install

要求:

  • 枕头
  • 火炬视觉
  • numpy公司

示例

^{pr2}$

当前特性

  • 在原始图像的范围内旋转和裁剪图像,以获得任何给定角度扰动(对于具有旋转噪声的训练样本)
  • 使用alpha通道支持在RGB/YUV/YCH颜色空间之间转换图像
  • 调整频道的对比度和亮度
  • 噪声发生

待办事项

这个项目仍在进行中。在

  • 随机轮作作物的均匀分布
  • 颜色查找表用于更快的颜色空间转换
  • 添加图像、文本、形状和像素化

项目页面

Github:https://github.com/paixi/ImageAug

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java使图像以正确的速度在屏幕上移动,以适应所有显示   内存Java分配:从预先存在/分配的池中分配对象   java这种书写方式?   Java正则表达式查找字符串的开头   java是否可以创建一个类来处理安卓中的所有日志代码(例如log.d(TAG,message))   如何使用Selenium和java单击WebTable任意页面上的WebElement   java解析字符串中的文件名   java刷新JTree内容   java如何覆盖RequestMappingHandler   爪哇数石头、布、剪刀赢了多少   struts中的java无效令牌   swing JTree,优化算法,Java   java Tomcat和SSL:密钥库格式无效