Skip to content

Commit 0d3cee1

Browse files
authored
Add all detection models to performance benchmark tests (#4266)
* Added all detection models to performance benchmark tests * removed yolo-v9 model from benchmark tests
1 parent f8f90d4 commit 0d3cee1

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/perf_v2/tasks/detection.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,16 @@
2121
MODEL_TEST_CASES = [
2222
ModelInfo(task=TASK_TYPE.value, name="atss_mobilenetv2", category="default"),
2323
ModelInfo(task=TASK_TYPE.value, name="yolox_s", category="speed"),
24+
ModelInfo(task=TASK_TYPE.value, name="dfine_x", category="accuracy"),
25+
ModelInfo(task=TASK_TYPE.value, name="atss_resnext101", category="other"),
26+
ModelInfo(task=TASK_TYPE.value, name="rtdetr_101", category="other"),
27+
ModelInfo(task=TASK_TYPE.value, name="rtdetr_18", category="other"),
28+
ModelInfo(task=TASK_TYPE.value, name="rtdetr_50", category="other"),
29+
ModelInfo(task=TASK_TYPE.value, name="rtmdet_tiny", category="other"),
30+
ModelInfo(task=TASK_TYPE.value, name="ssd_mobilenetv2", category="other"),
31+
ModelInfo(task=TASK_TYPE.value, name="yolox_tiny", category="other"),
32+
ModelInfo(task=TASK_TYPE.value, name="yolox_l", category="other"),
33+
ModelInfo(task=TASK_TYPE.value, name="yolox_x", category="other"),
2434
]
2535

2636
DATASET_TEST_CASES = [

0 commit comments

Comments
 (0)