未验证 提交 98939299 编写于 作者: J Jianfeng Wang 提交者: GitHub

fix(hub): fix detection models and links (#7)

上级 73b34503
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: BERT zh_CN: BERT
author: MegEngine Team author: MegEngine Team
tags: [nlp] tags: [nlp]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/nlp/bert
--- ---
```python ```python
......
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: DeepLabV3plus (VOC预训练权重) zh_CN: DeepLabV3plus (VOC预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision] tags: [vision]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/segmentation
--- ---
```python ```python
......
...@@ -6,14 +6,14 @@ summary: ...@@ -6,14 +6,14 @@ summary:
zh_CN: Faster-RCNN (COCO预训练权重) zh_CN: Faster-RCNN (COCO预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, detection] tags: [vision, detection]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/detection
--- ---
```python ```python
from megengine import hub from megengine import hub
model = hub.load( model = hub.load(
"megengine/models", "megengine/models",
"faster_rcnn_fpn_res50_coco_1x_800size", "faster_rcnn_res50_coco_1x_800size",
pretrained=True, pretrained=True,
use_cache=False, use_cache=False,
) )
...@@ -61,8 +61,9 @@ print(infer()) ...@@ -61,8 +61,9 @@ print(infer())
目前我们提供了retinanet的预训练模型, 在coco验证集上的结果如下: 目前我们提供了retinanet的预训练模型, 在coco验证集上的结果如下:
| model | mAP<br>@5-95 | | model | mAP<br>@5-95 |
| --- | --- | | --- | :---: |
| faster-rcnn-res50-1x-800size | 37.3 | | faster-rcnn-res50-1x-800size | 38.8 |
| faster-rcnn-res50-1x-800size-syncbn | 39.3 |
### 参考文献 ### 参考文献
...@@ -108,8 +109,9 @@ print(infer()) ...@@ -108,8 +109,9 @@ print(infer())
Currently we provide a `retinanet` model which is pretrained on `COCO2017` training set. The mAP on `COCO2017` val set can be found in following table. Currently we provide a `retinanet` model which is pretrained on `COCO2017` training set. The mAP on `COCO2017` val set can be found in following table.
| model | mAP<br>@5-95 | | model | mAP<br>@5-95 |
| --- | --- | | --- | :---: |
| faster-rcnn-res50-1x-800size | 37.3 | | faster-rcnn-res50-1x-800size | 38.8 |
| faster-rcnn-res50-1x-800size-syncbn | 39.3 |
### References ### References
......
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: MSPN(COCO 预训练权重) zh_CN: MSPN(COCO 预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, keypoints] tags: [vision, keypoints]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/keypoints
--- ---
```python3 ```python3
......
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: 深度残差网络(ImageNet 预训练权重) zh_CN: 深度残差网络(ImageNet 预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, classification] tags: [vision, classification]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/classification
--- ---
```python ```python
......
...@@ -6,15 +6,16 @@ summary: ...@@ -6,15 +6,16 @@ summary:
zh_CN: RetinaNet (COCO预训练权重) zh_CN: RetinaNet (COCO预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, detection] tags: [vision, detection]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/detection
--- ---
```python ```python
from megengine import hub from megengine import hub
model = hub.load( model = hub.load(
"megengine/models", "megengine/models",
"retinanet_res50_1x_800size", "retinanet_res50_coco_1x_800size",
pretrained=True, pretrained=True,
use_cache=False,
) )
model.eval() model.eval()
...@@ -60,8 +61,9 @@ print(infer()) ...@@ -60,8 +61,9 @@ print(infer())
目前我们提供了retinanet的预训练模型, 在coco验证集上的结果如下: 目前我们提供了retinanet的预训练模型, 在coco验证集上的结果如下:
| model | mAP<br>@5-95 | | model | mAP<br>@5-95 |
| --- | --- | | --- | :---: |
| retinanet-res50-1x-800size | 36.0 | | retinanet-res50-coco1x-800size | 36.4 |
| retinanet-res50-coco1x-800size-syncbn | 37.1 |
### 参考文献 ### 参考文献
...@@ -107,8 +109,9 @@ print(infer()) ...@@ -107,8 +109,9 @@ print(infer())
Currently we provide a `retinanet` model which is pretrained on `COCO2017` training set. The mAP on `COCO2017` val set can be found in following table. Currently we provide a `retinanet` model which is pretrained on `COCO2017` training set. The mAP on `COCO2017` val set can be found in following table.
| model | mAP<br>@5-95 | | model | mAP<br>@5-95 |
| --- | --- | | --- | :---: |
| retinanet-res50-1x-800size | 36.0 | | retinanet-res50-coco1x-800size | 36.4 |
| retinanet-res50-coco1x-800size-syncbn | 37.1 |
### References ### References
......
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: ShuffleNet V2(ImageNet 预训练权重) zh_CN: ShuffleNet V2(ImageNet 预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, classification] tags: [vision, classification]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/classification
--- ---
```python ```python
......
...@@ -6,7 +6,7 @@ summary: ...@@ -6,7 +6,7 @@ summary:
zh_CN: SimpleBaeline(COCO 预训练权重) zh_CN: SimpleBaeline(COCO 预训练权重)
author: MegEngine Team author: MegEngine Team
tags: [vision, keypoints] tags: [vision, keypoints]
github-link: https://github.com/megengine/models github-link: https://github.com/MegEngine/Models/tree/master/official/vision/keypoints
--- ---
```python3 ```python3
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册