onflow将图片文件制作为 OFRecord 数据集

最新推荐文章于 2024-08-09 23:22:37 发布

zjLOVEcyj

最新推荐文章于 2024-08-09 23:22:37 发布

阅读量129

点赞数

分类专栏： oneflow

本文链接：https://blog.csdn.net/cyj5201314/article/details/118966073

版权

python onflow ofrecord

将50张mnist图像和标签转为序列化数据写入ofrecord文件
images/train_set中保存50张mnist手写数字图像，28*28像素
iamges/train_label/label.txt中以json格式保存标签

# img2ofrecord.py
import cv2
import oneflow.core.record.record_pb2 as ofrecord
import six
import struct
import os
import argparse
import json

#将python内置类型数据转为Feature对象的函数，下同
def int32_feature(value):
    if not isinstance(value, (list, tuple)):
        value = [value]
    return ofrecord.Feature(int32_list=ofrecord.Int32List(value=value))


def int64_feature(value):
    if not isinstance(value, (list, tuple)):
        value = [value]
    return ofrecord.Feature(int64_list=ofrecord.Int64List(value=value))


def float_feature(value):
    if not isinstance(value, (list, tuple)):
        value = [value]
    return ofrecord.Feature(float_list=ofrecord.FloatList(value=value))


def double_feature(value):
    if not isinstance(value, (list, tuple