Skip to content

Commit b8032d0

Browse files
committed
docs(ssd): 调整文档结构,更新训练流程
1 parent 38b6373 commit b8032d0

File tree

6 files changed

+60
-28
lines changed

6 files changed

+60
-28
lines changed

docs/index.md

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,11 @@
2323
* 损失函数
2424
* `Hard Negative Mining`
2525
* `variance`变量
26-
* 训练
27-
* 执行命令
28-
* 训练日志
29-
* 测试日志
26+
* 具体实现
27+
* 命令行操作
28+
* 训练
29+
* 训练流程
30+
* 训练日志
31+
* 测试
32+
* 测试流程
33+
* 测试日志
File renamed without changes.

docs/ssd/index.md

Lines changed: 5 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,32 +1,18 @@
11

22
# SSD
33

4-
[SSD(Single Shot MultiBox Detector)](https://arxiv.org/abs/1512.02325)是一个`One-Stage`目标检测算法。其相对于之前的检查算法`YOLOv1、Faster RCNN`)而言,关键的区别在于
4+
[SSD(Single Shot MultiBox Detector)](https://arxiv.org/abs/1512.02325)是一个`One-Stage`目标检测算法。其相对于之前的检测算法`YOLOv1、Faster RCNN`)而言,关键的区别在于
55

66
1. 使用多个**不同尺度**的特征图进行边界框预测
77
2. 定义了一组**多个尺度、多个长宽比**的先验框
88
2. 利用**小卷积滤波器**计算特征图,以得到先验框偏移和分类成绩
99

10-
## 检测流程
10+
## 章节安排
1111

12-
1. 输入图像
13-
2. 通过卷积网络模型计算生成一组边界框偏移,以及和每个边界框对应的分类概率
14-
3. 结合先验框得到预测边界框
15-
4. 通过非最大抑制策略消除重叠的边界框
16-
5. 输出检测结果
12+
分两部分研究`SSD`算法:
1713

18-
## 训练流程
19-
20-
1. 加载训练数据
21-
2. 定义网络模型
22-
3. 定义损失函数
23-
4. 定义优化器
24-
5. 批量训练
25-
26-
下面操作中,默认定义
27-
28-
* 输入图像尺寸为$300\times 300$
29-
* 数据集为`PASCAL VOC`
14+
1. 原理解析
15+
2. 具体实现
3016

3117
## 相关阅读
3218

docs/ssd/train_process.md

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
2+
# 训练流程
3+
4+
* `py/train.py`
5+
6+
## 训练参数
7+
8+
* 数据集
9+
* 训练:`PASCAL VOC 07+12 trainval`
10+
* 测试:`PASCAL VOC 07 test`
11+
* 数据
12+
* 输入大小:`300`
13+
* 批量大小:`32`
14+
* 优化器:`SGD`
15+
* `lr: 1e-3`
16+
* `weight_decay: 5e-4`
17+
* 动量:`0.9`
18+
* 衰减器:`WarmupMultiStepLR`(自定义,`warmup + stepLR`
19+
* 起始学习率(`warmup_factor`):`1/3.0`
20+
* `warmup`迭代次数(`warmup_iters`):`500`
21+
* 衰减时刻:第`8`万次和第`10`万次迭代
22+
* 迭代总数:`12`万次
23+
* 模型
24+
* 基础网络:`VGG16`
25+
* 损失函数:`SmoothL1 Loss + Softmax Loss`
26+
27+
## 训练流程
28+
29+
* 批量载入图像
30+
* 图像预处理
31+
* 针对图像进行操作
32+
* 针对边界框进行操作:计算先验框;匹配先验框和标注框,转换边界框坐标格式
33+
* 模型计算
34+
* 损失计算
35+
* 梯度更新
36+
* 学习率更新

docs/ssd/using_command.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11

2-
# 执行命令
2+
# 命令行操作
33

44
执行训练、测试、检测命令如下
55

mkdocs.yml

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,13 @@ nav:
7373
- 损失函数: ssd/loss_function.md
7474
- 'Hard Negative Mining': ssd/hard_negative_mining.md
7575
- variance变量: ssd/variance.md
76-
- 训练:
77-
- 使用命令: ssd/using_command.md
78-
- 训练日志: ssd/train_res.md
79-
- 测试日志: ssd/test_res.md
76+
- 具体实现:
77+
- 命令行操作: ssd/using_command.md
78+
- 训练:
79+
- 训练流程: ssd/train_process.md
80+
- 训练日志: ssd/train_res.md
81+
- 测试:
82+
- 测试流程: ssd/test_process.md
83+
- 测试日志: ssd/test_res.md
84+
- 使用:
85+
- 使用日志: ssd/demo_res.md

0 commit comments

Comments
 (0)