MaitreChen / Cat_Dog_Classification

Are you a cat or a dog？😹 vs 🐕‍🦺这是一个基础的猫狗分类项目，采用Tensorflow框架构建，包含了训练、测试与推理部分~~💖

📣Introduction

一个非常easy的项目，仅仅只是为了通过猫狗分类实例熟悉基本的Tensorflow接口🎇

实验部分探索了三组不同的学习率对模型泛化性能的影响。

Furthermore，等之后兴致来潮就updating🤓~

💊Dependence

• python 3.8
• tensorflow 2.2.0
• tensorboardX 2.6
• Numpy 1.20.1
• matplotlib 3.7.1

✨Usage

Dataset

下载地址：https://www.kaggle.com/datasets/tongpython/cat-and-dog

训练集、验证集、测试集比例：8 : 1 : 1

目录结构：

data/
├── test
│   ├── cats
│   │   ├── cat.1.jpg
│   │   ├── cat.2.jpg
│   │   ├── .
│   │   ├── .
│   └── dogs
├── train
│   ├── cats
│   └── dogs
└── val
    ├── cats
    └── dogs

Checkpoints

链接：https://pan.baidu.com/s/13Sw-E-yE2xhlpP3TYwuTNg?pwd=3526 提取码：3526

🎉Result

Training Accuracy:95.59%
Validation Accuracy:84.40% 
Testing Accuracy:83.58%

👁‍🗨Reference

https://github.com/girishkuniyal/Cat-Dog-CNN-Classifier