Skip to content

GitLab

Commit bc245174 authored by dengjb's avatar dengjb
Browse files

Update README.md

parent f00d3a19
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ bc245174
...@@ -5,7 +5,7 @@ ...@@ -5,7 +5,7 @@
`Probabilistic two-stage detection`<br> `Probabilistic two-stage detection`<br>
[论文链接](https://arxiv.org/abs/2103.07461) [论文链接](https://arxiv.org/abs/2103.07461)
## 模型结构 ## 模型结构
Centernet对于输入图像(512,512,3)，下采样率为4，CenterNet将得到平面大小为(128,128)的输出. CenterNet的prediction head也由三个分支组成，分别为heatmap head，dimension head和offset head Centernet对于输入图像(512,512,3)，下采样率为4，CenterNet将得到平面大小为(128,128)的输出。CenterNet的prediction head也由三个分支组成，分别为heatmap head，dimension head和offset head。
![模型结构](resources%2Fimg.png) ![模型结构](resources%2Fimg.png)
## 算法原理 ## 算法原理
CenterNet是一种anchor free的目标检测算法,就是直接回归检测到的物体而不是回归anchors,不需要提前设定anchors CenterNet是一种anchor free的目标检测算法,就是直接回归检测到的物体而不是回归anchors,不需要提前设定anchors
...@@ -93,7 +93,7 @@ python demo/image_demo.py demo/demo.jpg ./configs/centernet/centernet-update_r50 ...@@ -93,7 +93,7 @@ python demo/image_demo.py demo/demo.jpg ./configs/centernet/centernet-update_r50
## result ## result
![demo.jpg](resources%2Fdemo.jpg) ![demo.jpg](resources%2Fdemo.jpg)
### 精度 ### 精度
| 模型名称 | batchsize | amp混精 | 精度 | | 模型名称 | batchsize | amp混精 | Box AP |
|:----------------:|:---------:|:-----:|:----:| |:----------------:|:---------:|:-----:|:----:|
| centernet-update | 16 | off | 40.1 | | centernet-update | 16 | off | 40.1 |
## 应用场景 ## 应用场景
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment