如何使用我自己的图像在tensorFlow中训练我的CNN神经网络

时间:2017-02-06 10:46:05

标签: tensorflow

我目前正在开发一个使用CNN张量流神经网络的程序,我想用我自己的图像进行训练和测试,请给我一些建议,因为我是深度学习的新手

感谢。

1 个答案:

答案 0 :(得分:0)

从python下载ronnie包。[{pythonpath} \ scripts \ pip3 ronnie] 将训练数据结构构造为[标签,图像像素]并执行以下代码以创建训练数据。

dataFileLoc = os.path.join(dir,“./ data / digitalRec / train.csv”) orgDF = collector.initial(dataFileLoc)

################################################## ######################### buildFormDF中的返回对象是dict结构中的ImgXs和LabelYs

{meta.DATA_INPUTXs:imgXs,meta.DATA_LABELYs:labelYs}

################################################## ######################## trainData = collector.buildFromDF(orgDF,batchNum = 1,batchSize = 30000)