update readme

README.md

@@ -1,7 +1,7 @@
 ## Classification：分类模型在Pytorch当中的实现
 ---
 
-### 目录
+## 目录
 1. [仓库更新 Top News](#仓库更新)
 2. [所需环境 Environment](#所需环境)
 3. [文件下载 Download](#文件下载)
@@ -16,10 +16,10 @@ BiliBili视频中的原仓库地址为：https://github.com/bubbliiiing/classifi
 
 **`2021-01`**:**仓库创建，支持模型训练，大量的注释，多个可调整参数。支持top1-top5的准确度评价。**   
 
-### 所需环境
+## 所需环境
 pytorch == 1.2.0
 
-### 文件下载
+## 文件下载
 训练所需的预训练权重都可以在百度云下载。     
 链接: https://pan.baidu.com/s/1Jxeyeni45PvGDuPNdhAjCw    
 提取码: uyke    
@@ -28,7 +28,7 @@ pytorch == 1.2.0
 链接: https://pan.baidu.com/s/1hYBNG0TnGIeWw1-SwkzqpA     
 提取码: ass8    
 
-### 训练步骤
+## 训练步骤
 1. datasets文件夹下存放的图片分为两部分，train里面是训练图片，test里面是测试图片。  
 2. 在训练之前需要首先准备好数据集，在train或者test文件里里面创建不同的文件夹，每个文件夹的名称为对应的类别名称，文件夹下面的图片为这个类的图片。文件格式可参考如下：
 ```
@@ -54,13 +54,13 @@ pytorch == 1.2.0
 4. 之后修改model_data文件夹下的cls_classes.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
@@ -94,7 +94,7 @@ _defaults = {
 img/cat.jpg
 ```  
 
-### 评估步骤
+## 评估步骤
 1. datasets文件夹下存放的图片分为两部分，train里面是训练图片，test里面是测试图片，在评估的时候，我们使用的是test文件夹里面的图片。  
 2. 在评估之前需要首先准备好数据集，在train或者test文件里里面创建不同的文件夹，每个文件夹的名称为对应的类别名称，文件夹下面的图片为这个类的图片。文件格式可参考如下：
 ```
@@ -146,6 +146,6 @@ _defaults = {
 ```
 5. 运行eval_top1.py和eval_top5.py来进行模型准确率评估。
 
-### Reference
+## Reference
 https://github.com/keras-team/keras-applications