weixin_47872887 2022-07-25 17:22 采纳率: 52.5%
浏览 25
已结题

yield 实现什么功能?

请问一下,这里的yied实现什么功能呢? 两个yield np.array(imgs_list), np.array(labels_list)都返回到哪里的呢?
可以不用yield么?

# 定义数据生成器,返回批次数据
def data_generator():
    imgs_list = []
    labels_list = []
    for i in index_list:
        # 将数据处理成希望的类型
        img = np.array(imgs[i]).astype('float32')
        label = np.array(labels[i]).astype('float32')
        imgs_list.append(img) 
        labels_list.append(label)
        if len(imgs_list) == BATCHSIZE:
            # 获得一个batchsize的数据,并返回
            yield np.array(imgs_list), np.array(labels_list)
            # 清空数据读取列表
            imgs_list = []
            labels_list = []

    # 如果剩余数据的数目小于BATCHSIZE,
    # 则剩余数据一起构成一个大小为len(imgs_list)的mini-batch
    if len(imgs_list) > 0:
        yield np.array(imgs_list), np.array(labels_list)
    return data_generator

  • 写回答

2条回答 默认 最新

  • 樱花的浪漫 人工智能领域优质创作者 2022-07-25 18:25
    关注

    return最为常见,十分简单,其返回运行一次函数所得到的结果,且在return语句之后python解释器会直接跳出函数,函数剩下的其它所有语句或者循环都不再被运行。
    yield是返回一个生成器(generator)对象。python中的生成器,可以使用next()来逐个获取yield返回的值。同时运行机制为在运行包含有生成器的函数的时候,只要碰到yield就暂停,这时候会保存当前运行的信息,也就是之前所产生的变量等都不会改变。
    yileld和return的区别就是一个函数不在运行,一个函数暂停运行,用next()方法继续运行,看自己的需求
    在上面的代码中,换为return,返回的是遍历完以后的整个运行结果

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 8月4日
  • 已采纳回答 7月27日
  • 创建了问题 7月25日

悬赏问题

  • ¥15 随身WiFi网络灯亮但是没有网络,如何解决?
  • ¥15 gdf格式的脑电数据如何处理matlab
  • ¥20 重新写的代码替换了之后运行hbuliderx就这样了
  • ¥100 监控抖音用户作品更新可以微信公众号提醒
  • ¥15 UE5 如何可以不渲染HDRIBackdrop背景
  • ¥70 2048小游戏毕设项目
  • ¥20 mysql架构,按照姓名分表
  • ¥15 MATLAB实现区间[a,b]上的Gauss-Legendre积分
  • ¥15 delphi webbrowser组件网页下拉菜单自动选择问题
  • ¥15 linux驱动,linux应用,多线程