【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∼

相关推荐
张槊哲7 分钟前
函数的定义与使用(python)
开发语言·python
船长@Quant12 分钟前
文档构建:Sphinx全面使用指南 — 实战篇
python·markdown·sphinx·文档构建
青松@FasterAI30 分钟前
【程序员 NLP 入门】词嵌入 - 上下文中的窗口大小是什么意思? (★小白必会版★)
人工智能·自然语言处理
AIGC大时代1 小时前
高效使用DeepSeek对“情境+ 对象 +问题“型课题进行开题!
数据库·人工智能·算法·aigc·智能写作·deepseek
硅谷秋水1 小时前
GAIA-2:用于自动驾驶的可控多视图生成世界模型
人工智能·机器学习·自动驾驶
多巴胺与内啡肽.1 小时前
深度学习--自然语言处理统计语言与神经语言模型
深度学习·语言模型·自然语言处理
偶尔微微一笑1 小时前
AI网络渗透kali应用(gptshell)
linux·人工智能·python·自然语言处理·编辑器
深度之眼1 小时前
2025时间序列都有哪些创新点可做——总结篇
人工智能·深度学习·机器学习·时间序列
晓数2 小时前
【硬核干货】JetBrains AI Assistant 干货笔记
人工智能·笔记·jetbrains·ai assistant
jndingxin2 小时前
OpenCV 图形API(60)颜色空间转换-----将图像从 YUV 色彩空间转换为 RGB 色彩空间函数YUV2RGB()
人工智能·opencv·计算机视觉