docs(yolo): 模型定义

zjZSTU · zjZSTU · commit 32a03782dbac · 2020-04-21T19:50:48.000+08:00
diff --git a/docs/imgs/figure-3.png b/docs/imgs/figure-3.png
diff --git a/docs/定义YOLO_v1模型.md b/docs/定义YOLO_v1模型.md
@@ -0,0 +1,13 @@
+
+# 定义YOLO_v1模型
+
+论文给出了一个`24`层卷积的`CNN`模型
+
+![](./imgs/figure-3.png)
+
+## 实现
+
+* `py/lib/models/basic_conv2d.py`
+* `py/lib/models/yolo_v1.py`
+
+模型输入`4`维数据（大小为`[N, C, H, W]`）；输出`3`维数据（大小为`[N, S*S, B*5+C]`)
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -32,4 +32,5 @@ nav:
     - Home: index.md
     - 架构解析: '架构解析.md'
     - 数据集: '数据集.md'
+    - 定义YOLO_v1模型: '定义YOLO_v1模型.md'
     - log: log.md
