Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
bubbliiiing committed Apr 20, 2022
1 parent fbac55c commit 8522d9a
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
## Classification:分类模型在Keras当中的实现
---

### 目录
## 目录
1. [仓库更新 Top News](#仓库更新)
2. [所需环境 Environment](#所需环境)
3. [文件下载 Download](#文件下载)
Expand All @@ -16,11 +16,11 @@ BiliBili视频中的原仓库地址为:https://github.com/bubbliiiing/classifi

**`2021-01`**:**仓库创建,支持模型训练,大量的注释,多个可调整参数。支持top1-top5的准确度评价。**

### 所需环境
## 所需环境
tensorflow-gpu==1.13.1
keras==2.1.5

### 文件下载
## 文件下载
训练所需的预训练权重都可以在百度云下载。
链接: https://pan.baidu.com/s/1MtnVfgZ9aN1bYfsBIGjkBA
提取码: eztx
Expand All @@ -29,7 +29,7 @@ keras==2.1.5
链接: https://pan.baidu.com/s/1hYBNG0TnGIeWw1-SwkzqpA
提取码: ass8

### 训练步骤
## 训练步骤
1. datasets文件夹下存放的图片分为两部分,train里面是训练图片,test里面是测试图片。
2. 在训练之前需要首先准备好数据集,在train或者test文件里里面创建不同的文件夹,每个文件夹的名称为对应的类别名称,文件夹下面的图片为这个类的图片。文件格式可参考如下:
```
Expand All @@ -55,13 +55,13 @@ keras==2.1.5
4. 在准备好数据集后,需要在根目录运行txt_annotation.py生成训练所需的cls_train.txt,运行前需要修改其中的classes_path,classes_path需要指向model_data下的txt文件,txt文件中是自己所要去区分的种类,将其修改成自己需要分的类。
5. 在train.py里面调整自己要选择的网络和权重后,就可以开始训练了!

### 预测步骤
#### a、使用预训练权重
## 预测步骤
### a、使用预训练权重
1. 下载完库后解压,model_data已经存在一个训练好的猫狗模型mobilenet025_catvsdog.h5,运行predict.py,输入
```python
img/cat.jpg
```
#### b、使用自己训练的权重
### b、使用自己训练的权重
1. 按照训练步骤训练。
2. 在classification.py文件里面,在如下部分修改model_path、classes_path、backbone和alpha使其对应训练好的文件;**model_path对应logs文件夹下面的权值文件,classes_path是model_path对应分的类,backbone对应使用的主干特征提取网络,alpha是当使用mobilenet的alpha值**
```python
Expand Down Expand Up @@ -95,7 +95,7 @@ img/cat.jpg
```


### 评估步骤
## 评估步骤
1. datasets文件夹下存放的图片分为两部分,train里面是训练图片,test里面是测试图片,在评估的时候,我们使用的是test文件夹里面的图片。
2. 在评估之前需要首先准备好数据集,在train或者test文件里里面创建不同的文件夹,每个文件夹的名称为对应的类别名称,文件夹下面的图片为这个类的图片。文件格式可参考如下:
```
Expand Down Expand Up @@ -146,5 +146,5 @@ _defaults = {
```
5. 运行eval_top1.py和eval_top5.py来进行模型准确率评估。

### Reference
## Reference
https://github.com/keras-team/keras-applications

0 comments on commit 8522d9a

Please sign in to comment.