![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
oneflow
zjLOVEcyj
composing code was a kind of art
展开
-
oneflow利用ofrecord数据集文件训练lenet模型
# lenet_train.pyimport numpy as npimport oneflow as flowimport oneflow.typing as tp#lenet网络结构def lenet(data, train=False): initializer = flow.truncated_normal(0.1) conv1 = flow.layers.conv2d( data, 32, 5, paddi原创 2021-07-22 10:04:30 · 107 阅读 · 0 评论 -
onflow将图片文件制作为 OFRecord 数据集
将50张mnist图像和标签转为序列化数据写入ofrecord文件images/train_set中保存50张mnist手写数字图像,28*28像素iamges/train_label/label.txt中以json格式保存标签# img2ofrecord.pyimport cv2import oneflow.core.record.record_pb2 as ofrecordimport siximport structimport osimport argparseimport.原创 2021-07-21 14:43:35 · 108 阅读 · 0 评论 -
oneflow制作ofrecord对象并保存ofrecord文件和从文件读取数据
import oneflow.core.record.record_pb2 as ofrecordimport siximport randomimport structdef int32_feature(value): if not isinstance(value, (list, tuple)): value = [value] return ofrecord.Feature(int32_list=ofrecord.Int32List(value=value.原创 2021-07-21 09:24:01 · 117 阅读 · 0 评论