File tree Expand file tree Collapse file tree 6 files changed +60
-28
lines changed Expand file tree Collapse file tree 6 files changed +60
-28
lines changed Original file line number Diff line number Diff line change 23
23
* 损失函数
24
24
* ` Hard Negative Mining `
25
25
* ` variance ` 变量
26
- * 训练
27
- * 执行命令
28
- * 训练日志
29
- * 测试日志
26
+ * 具体实现
27
+ * 命令行操作
28
+ * 训练
29
+ * 训练流程
30
+ * 训练日志
31
+ * 测试
32
+ * 测试流程
33
+ * 测试日志
File renamed without changes.
Original file line number Diff line number Diff line change 1
1
2
2
# SSD
3
3
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 ` )而言,关键的区别在于
5
5
6
6
1 . 使用多个** 不同尺度** 的特征图进行边界框预测
7
7
2 . 定义了一组** 多个尺度、多个长宽比** 的先验框
8
8
2 . 利用** 小卷积滤波器** 计算特征图,以得到先验框偏移和分类成绩
9
9
10
- ## 检测流程
10
+ ## 章节安排
11
11
12
- 1 . 输入图像
13
- 2 . 通过卷积网络模型计算生成一组边界框偏移,以及和每个边界框对应的分类概率
14
- 3 . 结合先验框得到预测边界框
15
- 4 . 通过非最大抑制策略消除重叠的边界框
16
- 5 . 输出检测结果
12
+ 分两部分研究` SSD ` 算法:
17
13
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 . 具体实现
30
16
31
17
## 相关阅读
32
18
Original file line number Diff line number Diff line change
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
+ * 学习率更新
Original file line number Diff line number Diff line change 1
1
2
- # 执行命令
2
+ # 命令行操作
3
3
4
4
执行训练、测试、检测命令如下
5
5
Original file line number Diff line number Diff line change 73
73
- 损失函数 : ssd/loss_function.md
74
74
- ' Hard Negative Mining ' : ssd/hard_negative_mining.md
75
75
- 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
You can’t perform that action at this time.
0 commit comments