Tensorflow学习：生成tfrecord文件

最新推荐文章于 2022-06-08 19:30:09 发布

空城不空99

最新推荐文章于 2022-06-08 19:30:09 发布

阅读量368

点赞数 1

分类专栏： Tensorflow学习文章标签： tensorflow 机器学习 python

本文链接：https://blog.youkuaiyun.com/cui1004/article/details/107869879

版权

Tensorflow学习：生成tfrecord文件

import tensorflow as tf
import os
import random
import math
import sys
from PIL import Image
import numpy as np

# 验证集数量
_NUM_TEST = 500

# 随机种子
_RANDOM_SEED = 0

# 数据集路径
DATASET_DIR = "D:/workspace/PyCharm/venv/cuiyongling/captcha/images/"

# tfrecord文件存放路径
TFRECORD_DIR = "D:/workspace/PyCharm/venv/cuiyongling/captcha/"


# 判断tfrecord文件是否存在
def _dataset_exists(dataset_dir):
    for split_name in ['train', 'test']:
        output_filename = os.path.join(dataset_dir, split_name + '.tfrecords')
        if not tf.gfile.Exists(output_filename):
            return False
    return True


# 获取所有验证码图片
def _get_filenames_and_classes(dataset_dir):
    photo_filenames = []
    for filename in os.listdir(dataset_dir):
        # 获取文件路径
        path = os.path.join(dataset_dir, filename)
        photo_filenames