当前位置:首页 > 文章列表 > 文章 > python教程 > Generator转NumPy优化图像像素随机化

Generator转NumPy优化图像像素随机化

2025-08-17 11:33:29 0浏览 收藏

小伙伴们有没有觉得学习文章很有意思?有意思就对了!今天就给大家带来《将Generator转NumPy数组优化图像像素随机化》,以下内容将会涉及到,若是在学习中对其中部分知识点有疑问,或许看了本文就能帮到你!

快速将Generator转换为NumPy数组并优化图像像素随机化

图像处理中,随机化像素顺序是一种常见的操作。原始代码中使用np.random.shuffle函数来实现,但效率较低。为了提升性能,开发者尝试使用生成器(generator)配合np.random.permutation,但遇到了类型转换的问题。本文将深入探讨如何利用np.random.permutation更高效地随机化图像像素,并解决将生成器转换为NumPy数组的难题。

使用 np.random.permutation 优化像素随机化

直接使用np.random.shuffle对大型数组进行原地洗牌效率较低。一个更高效的方法是生成一个随机排列的索引数组,然后使用该索引数组重新排列原始数组。

以下是改进后的代码示例:

import numpy as np
import time

def randomize_image(img):
    # convert image from (m,n,3) to (N,3)
    rndImg = np.reshape(img, (-1, img.shape[2]))
    np.random.shuffle(rndImg)
    rndImg = np.reshape(rndImg, img.shape)
    return rndImg

def randomize_image2(img):
    # convert image from (m,n,3) to (N,3)
    rndImg = np.reshape(img, (-1, img.shape[2]))
    i = np.random.permutation(len(rndImg))
    rndImg = rndImg[i, :]
    rndImg = np.reshape(rndImg, img.shape)
    return rndImg

# 示例用法
m, n = 1000, 1000
img = np.arange(m*n*3).reshape(m, n, 3)

start_time = time.perf_counter()
img1 = randomize_image(img)
end_time = time.perf_counter()
print('Time random shuffle: ', end_time - start_time)

start_time = time.perf_counter()
img2 = randomize_image2(img)
end_time = time.perf_counter()
print('Time random permutation: ', end_time - start_time)

在这个例子中,randomize_image2 函数使用了 np.random.permutation 生成一个随机索引数组 i。然后,它使用这个索引数组来重新排列 rndImg 数组的行,从而实现像素的随机化。

使用 NumPy Generator 进一步优化

从 NumPy 1.17 版本开始,NumPy 引入了新的随机数生成器 (Generator) API,它提供了更好的性能和更多的控制选项。可以利用它来进一步优化像素随机化。

import numpy as np

# 在函数外部初始化 Generator
rng = np.random.default_rng()

def randomize_image3(img):
    # convert image from (m,n,3) to (N,3)
    rndImg = np.reshape(img, (-1, img.shape[2]))
    i = rng.permutation(len(rndImg))
    rndImg = rndImg[i, :]
    rndImg = np.reshape(rndImg, img.shape)
    return rndImg

在这个例子中,rng.permutation 代替了 np.random.permutation。 注意: rng = np.random.default_rng() 应该在函数外部初始化,避免每次调用函数时都重新初始化,从而提高效率。

关于生成器 (Generator) 的说明

原始代码中尝试使用 yield 创建生成器,但这并不是解决问题的正确方向。生成器主要用于迭代产生值,而此处的目标是直接获得随机化后的NumPy数组。因此,不需要将函数改为生成器。

总结与注意事项

  • 使用 np.random.permutation 生成索引数组比直接使用 np.random.shuffle 更高效。
  • 利用 NumPy 的 Generator 对象可以进一步提升性能,特别是对于大型图像。
  • 避免不必要地使用生成器,因为它们并不适合直接生成NumPy数组的场景。
  • -1 在 np.reshape 中表示 "根据其他维度自动计算大小",使代码更简洁。

通过以上方法,可以显著提升图像像素随机化的效率,并更好地理解NumPy中随机数生成的相关工具。根据图像大小和具体应用场景,选择最合适的优化策略。

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于文章的相关知识,也可关注golang学习网公众号。

JavaScript异步编程演进全解析JavaScript异步编程演进全解析
上一篇
JavaScript异步编程演进全解析
JS物理引擎实现原理与技巧解析
下一篇
JS物理引擎实现原理与技巧解析
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    542次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    511次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    498次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
查看更多
AI推荐
  • 千音漫语:智能声音创作助手,AI配音、音视频翻译一站搞定!
    千音漫语
    千音漫语,北京熠声科技倾力打造的智能声音创作助手,提供AI配音、音视频翻译、语音识别、声音克隆等强大功能,助力有声书制作、视频创作、教育培训等领域,官网:https://qianyin123.com
    191次使用
  • MiniWork:智能高效AI工具平台,一站式工作学习效率解决方案
    MiniWork
    MiniWork是一款智能高效的AI工具平台,专为提升工作与学习效率而设计。整合文本处理、图像生成、营销策划及运营管理等多元AI工具,提供精准智能解决方案,让复杂工作简单高效。
    190次使用
  • NoCode (nocode.cn):零代码构建应用、网站、管理系统,降低开发门槛
    NoCode
    NoCode (nocode.cn)是领先的无代码开发平台,通过拖放、AI对话等简单操作,助您快速创建各类应用、网站与管理系统。无需编程知识,轻松实现个人生活、商业经营、企业管理多场景需求,大幅降低开发门槛,高效低成本。
    190次使用
  • 达医智影:阿里巴巴达摩院医疗AI影像早筛平台,CT一扫多筛癌症急慢病
    达医智影
    达医智影,阿里巴巴达摩院医疗AI创新力作。全球率先利用平扫CT实现“一扫多筛”,仅一次CT扫描即可高效识别多种癌症、急症及慢病,为疾病早期发现提供智能、精准的AI影像早筛解决方案。
    195次使用
  • 智慧芽Eureka:更懂技术创新的AI Agent平台,助力研发效率飞跃
    智慧芽Eureka
    智慧芽Eureka,专为技术创新打造的AI Agent平台。深度理解专利、研发、生物医药、材料、科创等复杂场景,通过专家级AI Agent精准执行任务,智能化工作流解放70%生产力,让您专注核心创新。
    212次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码