【huggingface】【pytorch-image-models】timm框架中使用albumentations库数据增广

文章目录

  • 一、前言
  • 二、实操
    • [2.1 声明库](#2.1 声明库)
    • [2.2 定义你的数据增广算子](#2.2 定义你的数据增广算子)
    • [2.3 加入其中](#2.3 加入其中)

一、前言

问题是这样的,在使用timm框架训练时,发现数据增广不够,想用Albumentations库的数据增广,怎么把后者嵌入到前者的训练中。

其实也是比较简单的,这里笔者也是做个笔记以往忘记,需要的童鞋自取。

二、实操

我们知道训练才需要数据增广,原始的数据增广是在:

其实也就是用XXX_tfl来收集数据增广的算子,

那么要加入Albumentations的算子,只需要3步:

2.1 声明库

py 复制代码
import numpy as np
from PIL import Image
import albumentations as A

2.2 定义你的数据增广算子

py 复制代码
albumentations_transform = A.Compose([
    A.XXXXX()
])

2.3 加入其中

py 复制代码
primary_tfl += [
    transforms.Lambda(lambda img: Image.fromarray(
        albumentations_transform_1(image=np.array(img))['image'].astype('uint8')))
]

这样在最后跟原始的数据增广算子,用transforms.Compose(XX)返回即可,这个本身自带了。

以上就可以搞定,在根据我之前的博客打印出预处理以后的图片即可。Enjoy~

∼ O n e p e r s o n g o f a s t e r , a g r o u p o f p e o p l e c a n g o f u r t h e r ∼ \sim_{One\ person\ go\ faster,\ a\ group\ of\ people\ can\ go\ further}\sim ∼One person go faster, a group of people can go further∼

相关推荐
数据与人工智能律师1 分钟前
数字资产革命中的信任之锚:RWA法律架构的隐形密码
大数据·网络·人工智能·云计算·区块链
CHANG_THE_WORLD13 分钟前
封装一个png的编码解码操作
图像处理·人工智能·计算机视觉
赛丽曼26 分钟前
Assistant API的原理及应用
人工智能·chatgpt
菜包eo1 小时前
二维码驱动的独立站视频集成方案
网络·python·音视频
Yo_Becky1 小时前
【PyTorch】PyTorch预训练模型缓存位置迁移,也可拓展应用于其他文件的迁移
人工智能·pytorch·经验分享·笔记·python·程序人生·其他
DeepSeek-大模型系统教程1 小时前
深入金融与多模态场景实战:金融文档分块技术与案例汇总
人工智能·ai·语言模型·程序员·大模型·大模型学习·大模型教程
xinxiangwangzhi_1 小时前
pytorch底层原理学习--PyTorch 架构梳理
人工智能·pytorch·架构
yzx9910131 小时前
关于网络协议
网络·人工智能·python·网络协议
AiTEN_Robot1 小时前
AGV 无人叉车关键技术问题解析:精准定位算法 / 安全避障逻辑 / 系统对接协议全方案
人工智能·机器人·自动化·制造
fangeqin1 小时前
ubuntu源码安装python3.13遇到Could not build the ssl module!解决方法
linux·python·ubuntu·openssl