# CNN-AlexNet

**Repository Path**: deathvader/cnn-alex-net

## Basic Information

- **Project Name**: CNN-AlexNet
- **Description**: 使用python+tensorflow 2实现了CNN经典模型AlexNet
- **Primary Language**: Python
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 0
- **Created**: 2021-09-05
- **Last Updated**: 2022-06-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: Python, tensorflow2, TensorFlow, CNN, Deep-learning

## README

# AlexNet

#### 介绍
​		复现AlexNet经典网络架构, 使用的数据集为Kaggle上面的一个鸟类数据集,包含有39364张训练图像，1375张验证图像和1375张测试图像（每种鸟类五张）。数据集下载地址: https://www.kaggle.com/gpiosenka/100-bird-species
(后续该数据集有更新,请参照下图设置文件结构)

#### 软件架构
主体使用python和tensorflow实现


#### 安装教程

1.  Python 3.7 
2.  TensorFlow(gpu) 2.1.0 注意与python版本必须匹配
3.  Numpy
4.  Pandas

#### 使用说明

1.  本项目使用断点续训方式, 检查点和数据保存文件夹为checkpoint文件夹。项目主体不带有训练好的数据,请先运行train_model.py训练模型。
模型训练的参数保存在checkpoint文件夹里面
2.  注意修改文件路径，否则报错
3.  Core中的utils工具包是用来生成一个验证xlsx文件的(无需手动运行)，注意不要更改原数据集的文件结构

![输入图片说明](https://images.gitee.com/uploads/images/2021/1106/155638_6c34feaa_5083077.png "屏幕截图.png")

4.  训练好后使用Predict_app.py运行图像分辨(输入是图片的绝对路径)
![输入图片说明](https://images.gitee.com/uploads/images/2021/1106/160122_f6a7aeb5_5083077.png "屏幕截图.png")