猫狗识别

第1关：猫狗识别

import numpy as np

import pandas as pd

import tensorflow as tf

from tensorflow import keras

import os

import matplotlib.pyplot as plt

import matplotlib.image as img

import seaborn as sns

'''setting seed'''

seed = 0

np.random.seed(seed)

tf.random.set_seed(3)

TRAIN_DIR_PATH = r'/data/workspace/myshixun/animal/train'

file_names = os.listdir(TRAIN_DIR_PATH)

print('There are {} number of images in directory.'.format(len(file_names)))

def to_dataframe(file_names):

files, labels = list(), list()

for file in file_names:

files.append(file)

labels.append(file[:3])

df = pd.DataFrame({'filename':files, 'label':labels})

return df

df = to_dataframe(file_names)

df.head()

print('data set label distribution:n',df['label'].value_counts())