本文将一个完整的tf2.0框架下使用CNN模型解决图像分类问题 喜欢记得关注我 点收藏不迷路 辛苦整理免费分享的
import glob
import os
import cv2
import numpy as np
import random
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import losses,layers,optimizers
from tensorflow.keras.callbacks import EarlyStopping
tf.random.set_seed(2222)
np.random.seed(2222)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
assert tf.__version__.startswith('2.')
def Data_Generation():
X_data=[];Y_data=[]
path_data=[];path_label=[]
#path_file=os.getcwd()
files=os.listdir('pokemon')
for file in files:
print(file)
for path in glob.glob('pokemon/'+file+'/*.*'):
if 'jpg' or 'png' or 'jpeg' in path:
path_data.append(path)
random.shuffle (path_data) #打乱数据
for paths in path_data:
if '