猫狗识别

---

第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)
#####Begin#####
TRAIN_DIR_PATH = r'/data/workspace/myshixun/animal/train'
#####End#####
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())