diff --git a/README.md b/README.md index 1352e5f..78e995d 100644 --- a/README.md +++ b/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