import tensorflow as tf
import os
# from tensorflow.keras.layers import
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense,Conv2D,Flatten,Dropout,MaxPooling2D
import matplotlib.pyplot as plt
import cv2
URL = 'https://storage.googleapis.com/mledu-datasets/cats_and_dogs_filtered.zip'
path_to_zip = tf.keras.utils.get_file("cats_and_dogs.zip",origin=URL,extract=True)
PATH = os.path.join(os.path.dirname(path_to_zip),"cats_and_dogs_filtered")#C:\Users\Administrator\.keras\datasets\cats_and_dogs_filtered
print(PATH)
train_dir = os.path.join(PATH,"train")
validation_dir = os.path.join(PATH,"validation")
train_cats_dir = os.path.join(train_dir,"cats")
train_dogs_dir = os.path.join(train_dir,"dogs")
validation_cats_dir = os
tensorflow实现猫狗分类及单张图片测试
最新推荐文章于 2023-11-28 19:26:57 发布