0 数据
说明:忽略数据的合理性,本数据为测试代码使用的。
1 data_process.py
import tensorflow as tf
import numpy as np
import os
import cv2
def read_imgs_to_arr(imgs_path):
imgs = []
for img_path in imgs_path:
img = cv2.imread(img_path)
imgs.append(np.array(img) / 255.0)
return imgs
def get_ImgsAndLabel(data_path):
imgs_name = []
labels_name = []
for label_name in os.listdir(data_path):
for img_name in os.listdir(data_path + label_name):
if os.path.isfile(data_path + label_name + '/' + img_name):
imgs_name.a