2021
04-07
04-07
tensorflow学习笔记之tfrecord文件的生成与读取
训练模型时，我们并不是直接将图像送入模型，而是先将图像转换为tfrecord文件，再将tfrecord文件送入模型。为进一步理解tfrecord文件，本例先将6幅图像及其标签转换为tfrecord文件，然后读取tfrecord文件，重现6幅图像及其标签。1、生成tfrecord文件importosimportnumpyasnpimporttensorflowastffromPILimportImagefilenames=['images/cat/1.jpg','images/cat/2.jpg','images/dog/1.jpg','images/dog/2.jpg','images/p...
继续阅读 >