使用albumentations库进行数据增强

2023-11-08

需要注意增强之后以字典键值的形式调用输出,且可以保证img、gt的随机增强是一一对应一致的。

import albumentations as albu

tfrd = albu.Compose([
    albu.HorizontalFlip(p=0.5),
    albu.VerticalFlip(p=0.5),
    albu.ShiftScaleRotate(shift_limit = 0,scale_limit = 0,rotate_limit=(-30,30),p=1,border_mode=cv2.BORDER_CONSTANT,value=0),
    albu.Blur(blur_limit=3, p=0.3),
])

for name in trainlist:
    img = cv2.imread(os.path.join(img_dir,name))
    gt = cv2.imread(os.path.join(gt_dir,name))
    
    for j in range(4):
        ns = tfrd(image=img,mask=gt)
        newimg,newgt= ns['image'],ns['mask']
        cv2.imwrite(os.path.join(img_dir,'t'+str(j)+'_'+name),newimg)
        cv2.imwrite(os.path.join(gt_dir,'t'+str(j)+'_'+name),newgt)
        tmp.append('t'+str(j)+'_'+name)

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

使用albumentations库进行数据增强 的相关文章

随机推荐

  • nodejs(7)---中间件的使用

    中间件相当于过滤器模式 当前端发送信息到后端的时候 先经过中间件进行验证信息 再到后台调用接口获取数据 我们看看node中中间件怎么使用 一个中间件的完整形式是这样的 function demo middleware err req res
  • MySQL数据库的异步写入

    注意 数据库pymysql的commit 和execute 在提交数据时 都是同步提交至数据库 由于scrapy框架数据的解析和异步多线程的 所以scrapy的数据解析速度 要远高于数据的写入数据库的速度 如果数据写入过慢 会造成数据库写入
  • 回文序列 c++实现

    目录 题目 代码 解法1思路 解法2思路 题目 回文序列指的是从前往后遍历得到的序列和从后往前遍历完全相同的序列 现在给你n个整数a1 a2 an 需要你判断这个序列是否为回文序列 是输出1 不是输出0 代码 include
  • 树莓派搭建Obsidian LiveSync同步数据

    树莓派搭建Obsidian LiveSync同步数据 一 Docker安装couchdb数据库 在 usr local中创建couchdb文件夹 mkdir couchdb 在couchdb文件夹中创建local ini文件 touch l
  • Flutter框架详解

    跨平台框架常识 单页面应用 跨平台框架的设计理念 跨平台框架的架构 Flutter 核心架构 Flutter Framework dart ui library Rendering library Widget library Materi
  • 学习 Rust cookbook 之算法篇(algorithm)part 2

    原文作者 suhanyujie 永久链接 https github com suhanyujie rust cookbook note 博客链接 https ishenghuo cnblogs com 学习 Rust cookbook 之算
  • 计算机word文档无法工作,win10所有word文档都打不开如何解决_win10电脑所有word文档无法打开解决教程...

    word是我们经常会用到的文档编辑工具 用户们都会在电脑中安装使用 不过使用的时候难免遇到一些问题 比如有win10 ghost 纯净版系统用户要打开word文档的时候发现电脑中所有word文档都打不开了 这是什么情况呢 为此 win7之家
  • uniapp 原生puls和uniapp权限授权 代码

    我们可以用plus里面的方法打包App 开启权限 也可以通过uniapp里面的App权限设置开始权限授权 下面就简单介绍 代码的使用
  • Java lambda 使用index

    public static
  • 3d数学之向量详解

    向量的定义 既有大小又有方向的量称之为向量 与之相对应的是标量 标量是只有大小没有方向的量 一个向量的一般在头上添加一个箭头表示 比如向量V 可以表示为 V vec V V 游戏中一般以二维向量跟三维向量居多 例如一个由A点指向B点的向量
  • erp仓库管理系统,适合各种行业的仓库,有供应商管理 ,商品管理

    erp仓库管理系统 适合各种行业的仓库 有供应商管理 商品管理 客户管理 销售管理 收款管理 付款管理 销售明细管理 采购明细管理 多仓库调拨管理4110622654117171凤皇培训学校
  • 晶体(crystal、无源晶振)两端电容取值计算

    数字电路中往往会需要一个时钟源 常用的比如晶体振荡器 oscillator 有源晶振 晶体 crystal 无源晶振 等 一般晶体用的更多些 对于晶体来说 一般电路设计时会在晶体两端各接一个电容到地 如下图所示 上面电路中晶体两端的电容是用
  • D类功率放大器

    深度剖析D类功率放大 http DXdlgf 我们知道 功率放大器一般分为甲类 乙类 甲乙类 丙类 丁类 即D类 和戊类等多种工作方式 为了提高功率和效率 一般的方法是降低三极管的静态工作点及由甲类到乙类 甚至到丙类 甲类 乙类 甲乙类的工
  • cublasCgetrfBatched cublasCgetriBatched sample

    This example demonstrates how to use the CUBLAS library by scaling an array of floating point values on the device and c
  • Mysql第三章:存储引擎(MyISAM和Innodb)

    目录 1 什么是存储引擎 2 常用存储引擎 2 1 MyISAM特点 2 2 Innodb特征 2 2 1 Innodb优点 2 2 2 innoDB内存模型 3 MVCC多版本控制器 3 1 首先了解事务ACID 的实现 3 2 首先了解
  • Nginx的配置与部署及例子

    Nginx的配置与部署及例子 1 Nginx程序包 目前最新的开发版本时1 1 12 Linux Unix nginx 1 1 12 tar gz Windows nginx 1 1 12 zip 我们可以下载稳定版尝试 Linux Uni
  • UE4/UE5 使用Unreal Datasmith完全教程

    插件整理下载 点击下载 UE4 4 19 UE5 5 3 官方 入口 Datasmith支持导入Twinmotion Twinmotion教程汇总 2021 4 2 Datasmith安装到4 20 4 23版本 UE4 在4 20 23版
  • iis ajax json,IIS配置jquery ajax、 json,mp4文件支持

    11 条件加载Jquery 如果CDN没有下载到Jquery 则从本地读取 在MIME增加一个Json类型 在MIME增加一个配置如 添加 gt 文件扩展名为 JSON MIME类型为text json 也有将application x j
  • 深度学习的归一化方法

    目录 1 为什么使用 Normalization 2 BatchNorm vs LayerNorm 适用场景 工作原理 3 GroupNorm 本次介绍一下各种归一化方法 包括 BatchNorm LayerNorm InstanceNor
  • 使用albumentations库进行数据增强

    需要注意增强之后以字典键值的形式调用输出 且可以保证img gt的随机增强是一一对应一致的 import albumentations as albu tfrd albu Compose albu HorizontalFlip p 0 5