基于ResNet-50的猫狗分类训练

导入库：

import os

import numpy as np

import pandas as pd

import tensorflow as tf

from tensorflow.keras.preprocessing.image import ImageDataGenerator

from tensorflow.keras.layers import Dense, Dropout

from sklearn.model_selection import train_test_split

import matplotlib.pyplot as plt

import shutil

import tqdm

# 导入数据集

CONTENT_DIR = '/G：'

TRAIN_DIR = CONTENT_DIR + '/train'

VALID_DIR = CONTENT_DIR + '/valid'

if not os.path.exists(CONTENT_DIR):

    import zipfile

    with zipfile.ZipFile('E:/6.10dogcat/train.zip', 'r') as zipf:

        zipf.extractall(CONTENT_DIR)

    img_filenames = os.listdir(TRAIN_DIR)

    dog_filenames = [fn for fn in img_filenames if fn.startswith('dog')]

    cat_filenames = [fn for fn in img_filenames if fn.startswith('cat')]

    dataset_filenames = train_test_split(

        dog_filenames, cat_filenames, test_size=0.1, shuffle=True, random_state=42

    )