很多同学(针对新手)在训练mnist数据的时候,根据书本上的内容都可以很好很快的编辑并跑出来,但是一旦换成自己的文件夹,就很头疼,毕竟mnist里面一个read_data解决你所有的输入问题,然而在现实中,该read_data是要自己编辑的,本文主要针对非ont_hot数据,如何利用tensorflow搭起网络并跑通自己的数据,话不多说,直接上代码。
python版本:2.7
tensorflow 版本:1.1.0
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Thu Jan 25 11:28:55 2018
@author:huangxd
"""
"""
vision:python3
author:huangxd
"""
import os
import math
import numpy as np
import tensorflow as tf
#生成图片路径和标签list
#train_dir='C:/Users/hxd/Desktop/tensorflow_study/Alexnet_dr'
zeroclass = []
label_zeroclass = []
oneclass = []
label_oneclass = []
twoclass = []
label_twoclass = []
threeclass = []
label_threeclass = []
fourclass = []
label_fourclass = []
fiveclass = []
label_fivec