Tensorflow 2.0—数据集加载
Tensorflow 2.0—数据集加载
tf.data.Dataset API
import tensorflow as tf
import numpy as np
import os
import PIL.Image as Image
import json
def create_dataset(base_dir):
CATE_GROY ={'hand_small': 1, 'hand_large':2}
name_list = [os.path.splitext(i)[0] for
原创
2020-08-14 23:10:52 ·
500 阅读 ·
0 评论