Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
HDETR_pytorch
Commits
7ef68522
Commit
7ef68522
authored
Apr 14, 2025
by
chenzk
Browse files
Update url.md
parent
e9096d8a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
12 deletions
+4
-12
README.md
README.md
+4
-12
No files found.
README.md
View file @
7ef68522
...
@@ -69,15 +69,7 @@ pip install mmdet==2.26.0 (对应mmcv 1.7.1)
...
@@ -69,15 +69,7 @@ pip install mmdet==2.26.0 (对应mmcv 1.7.1)
```
```
## 数据集
## 数据集
通过SCNet下载
[
COCO2017
](
http://113.200.138.88:18080/aidatasets/coco2017
)
所需数据集:
通过SCNet下载
[
COCO2017
](
https://cocodataset.org/#home
)
所需数据集。
[
训练数据
](
http://113.200.138.88:18080/aidatasets/coco2017/-/blob/main/train2017.zip
)
[
验证数据
](
http://113.200.138.88:18080/aidatasets/coco2017/-/blob/main/val2017.zip
)
[
测试数据
](
http://113.200.138.88:18080/aidatasets/coco2017/-/blob/main/test2017.zip
)
[
标签数据
](
http://113.200.138.88:18080/aidatasets/coco2017/-/blob/main/annotations_trainval2017.zip
)
数据集的目录结构如下:
数据集的目录结构如下:
...
@@ -132,7 +124,7 @@ bash ./train_multi.sh
...
@@ -132,7 +124,7 @@ bash ./train_multi.sh
```
bash
```
bash
python test.py
--pre_trained_model
<checkpoint path>
--coco_path
<coco path>
python test.py
--pre_trained_model
<checkpoint path>
--coco_path
<coco path>
```
```
以
[
r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
](
http://
113.200.138.88:18080/aimodels/findsource-dependency/hdetr_pytorch/-/blob/main
/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
)
模型为例:
以
[
r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
](
http
s
://
github.com/HDETR/H-Deformable-DETR/releases/download/v0.1
/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
)
模型为例:
```
bash
```
bash
python test.py
--pre_trained_model
/path/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
--coco_path
/path/coco/data
--two_stage
--num_queries_one2many
1500
--mixed_selection
--look_forward_twice
python test.py
--pre_trained_model
/path/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pth
--coco_path
/path/coco/data
--two_stage
--num_queries_one2many
1500
--mixed_selection
--look_forward_twice
```
```
...
@@ -150,7 +142,7 @@ COCO2017测试集上的单张图像结果展示:
...
@@ -150,7 +142,7 @@ COCO2017测试集上的单张图像结果展示:
</div>
</div>
### 精度
### 精度
测试数据:
[
COCO-test2017
](
http://113.200.138.88:18080/aidatasets/coco2017/-/blob/main/test2017.zip
)
,测试模型config:
[
r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh
](
./configs/two_stage/deformable-detr-hybrid-branch/12eps/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh
)
测试数据:COCO-test2017,测试模型config:
[
r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh
](
./configs/two_stage/deformable-detr-hybrid-branch/12eps/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh
)
结果如下表所示:
结果如下表所示:
| Device | Backbone | query | epochs | AP |
| Device | Backbone | query | epochs | AP |
...
@@ -166,7 +158,7 @@ COCO2017测试集上的单张图像结果展示:
...
@@ -166,7 +158,7 @@ COCO2017测试集上的单张图像结果展示:
网安,交通,政府
网安,交通,政府
## 预训练权重
## 预训练权重
-
[
SCNet AIModels - hdetr_pytorch
](
http://
113.200.138.88:18080/aimodels/findsource-dependency/hdetr_pytorc
h
)
-
[
r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps
](
http
s
://
github.com/HDETR/H-Deformable-DETR/releases/download/v0.1/r50_hybrid_branch_lambda1_group6_t1500_dp0_mqs_lft_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage_12eps.pt
h
)
## 源码仓库及问题反馈
## 源码仓库及问题反馈
-
https://developer.hpccube.com/codes/modelzoo/hdetr_pytorch
-
https://developer.hpccube.com/codes/modelzoo/hdetr_pytorch
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment