Commit bc245174 authored by dengjb's avatar dengjb
Browse files

Update README.md

parent f00d3a19
......@@ -5,7 +5,7 @@
`Probabilistic two-stage detection`<br>
[论文链接](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)
## 算法原理
CenterNet是一种anchor free的目标检测算法,就是直接回归检测到的物体而不是回归anchors,不需要提前设定anchors
......@@ -93,7 +93,7 @@ python demo/image_demo.py demo/demo.jpg ./configs/centernet/centernet-update_r50
## result
![demo.jpg](resources%2Fdemo.jpg)
### 精度
| 模型名称 | batchsize | amp混精 | 精度 |
| 模型名称 | batchsize | amp混精 | Box AP |
|:----------------:|:---------:|:-----:|:----:|
| 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