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

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 10:25
    关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
    weixin_47872887 2022-07-26 00:37

    可否简单的理解,上文的yield一批一批的返回一个batch数据?

    回复
查看更多回答(1条)
编辑
预览

报告相同问题?

问题事件

  • 系统已结题 8月3日
  • 已采纳回答 7月27日
  • 创建了问题 7月25日
手机看
程序员都在用的中文IT技术交流社区

程序员都在用的中文IT技术交流社区

专业的中文 IT 技术社区,与千万技术人共成长

专业的中文 IT 技术社区,与千万技术人共成长

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

关注【CSDN】视频号,行业资讯、技术分享精彩不断,直播好礼送不停!

客服 返回
顶部