当前位置:   article > 正文

tensorflow 搭建神经网络框架_activation='relu',kernel_initializer=tf.truncated_

activation='relu',kernel_initializer=tf.truncated_normal_initializer(stddev=

1)准备数据,打开文件夹读取数据

  1. def read_data(data_dir):
  2. datas = []
  3. labels = []
  4. for fname in os.listdir(data_dir):
  5. fpath = os.path.join(data_dir, fname)
  6. image = Image.open(fpath)
  7. image = image.resize((32,32))
  8. data = np.array(image) / 255.0
  9. label = int(fname.split("_")[0])
  10. datas.append(data)
  11. labels.append(label)
  12. datas = np.array(datas)
  13. labels = np.array(labels)
  14. return fpaths, datas, labels

这里,我们从data_dir文件夹中读取文件列表,并且默认c_name.jpg为图片文件的格式,c是类别,name是图片名字。

读取完数据之后,把图片数据传入datas 中,类别信息传入labels 中,且datas与labels一一对应。

2)打乱读入的顺序,即改变datas与labels中数据的先后的顺序(或者进行其他处理)

  1. num_example=data.shape[0]
  2. arr=np.arange(num_example)
  3. np.random.shuffle(arr)
  4. datas=datas[arr]
  5. labels=labels[arr]
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/306187
推荐阅读
相关标签
  

闽ICP备14008679号