Skip to content

Commit

Permalink
[Feature] Add support to RT-DETRv2 model
Browse files Browse the repository at this point in the history
  • Loading branch information
CVHub520 committed Jul 29, 2024
1 parent 3e388b4 commit 0fe5cff
Show file tree
Hide file tree
Showing 10 changed files with 627 additions and 8 deletions.
10 changes: 10 additions & 0 deletions anylabeling/configs/auto_labeling/models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,16 @@
config_file: ":/ram_plus_swin_large_14m.yaml"
- model_name: "ram_swin_large_14m-r20231024"
config_file: ":/ram_swin_large_14m.yaml"
- model_name: "rtdetrv2l-r20240730"
config_file: ":/rtdetrv2l.yaml"
- model_name: "rtdetrv2m-r20240730"
config_file: ":/rtdetrv2m.yaml"
- model_name: "rtdetrv2m7x-r20240730"
config_file: ":/rtdetrv2m7x.yaml"
- model_name: "rtdetrv2s-r20240730"
config_file: ":/rtdetrv2s.yaml"
- model_name: "rtdetrv2x-r20240730"
config_file: ":/rtdetrv2x.yaml"
- model_name: "rtdetr_r50-r20230520"
config_file: ":/rtdetr_r50.yaml"
- model_name: "rtmdet_m_coco_person_rtmo_m-r20240112"
Expand Down
86 changes: 86 additions & 0 deletions anylabeling/configs/auto_labeling/rtdetrv2l.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: rtdetrv2
name: rtdetrv2l-r20240730
display_name: RT-DETRv2-L PaddleDetection
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.4.0/rtdetrv2_r50vd_6x_coco.onnx
score_threshold: 0.45
classes:
- person
- bicycle
- car
- motorcycle
- airplane
- bus
- train
- truck
- boat
- traffic light
- fire hydrant
- stop sign
- parking meter
- bench
- bird
- cat
- dog
- horse
- sheep
- cow
- elephant
- bear
- zebra
- giraffe
- backpack
- umbrella
- handbag
- tie
- suitcase
- frisbee
- skis
- snowboard
- sports ball
- kite
- baseball bat
- baseball glove
- skateboard
- surfboard
- tennis racket
- bottle
- wine glass
- cup
- fork
- knife
- spoon
- bowl
- banana
- apple
- sandwich
- orange
- broccoli
- carrot
- hot dog
- pizza
- donut
- cake
- chair
- couch
- potted plant
- bed
- dining table
- toilet
- tv
- laptop
- mouse
- remote
- keyboard
- cell phone
- microwave
- oven
- toaster
- sink
- refrigerator
- book
- clock
- vase
- scissors
- teddy bear
- hair drier
- toothbrush
86 changes: 86 additions & 0 deletions anylabeling/configs/auto_labeling/rtdetrv2m.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: rtdetrv2
name: rtdetrv2m-r20240730
display_name: RT-DETRv2-M PaddleDetection
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.4.0/rtdetrv2_r34vd_120e_coco.onnx
score_threshold: 0.45
classes:
- person
- bicycle
- car
- motorcycle
- airplane
- bus
- train
- truck
- boat
- traffic light
- fire hydrant
- stop sign
- parking meter
- bench
- bird
- cat
- dog
- horse
- sheep
- cow
- elephant
- bear
- zebra
- giraffe
- backpack
- umbrella
- handbag
- tie
- suitcase
- frisbee
- skis
- snowboard
- sports ball
- kite
- baseball bat
- baseball glove
- skateboard
- surfboard
- tennis racket
- bottle
- wine glass
- cup
- fork
- knife
- spoon
- bowl
- banana
- apple
- sandwich
- orange
- broccoli
- carrot
- hot dog
- pizza
- donut
- cake
- chair
- couch
- potted plant
- bed
- dining table
- toilet
- tv
- laptop
- mouse
- remote
- keyboard
- cell phone
- microwave
- oven
- toaster
- sink
- refrigerator
- book
- clock
- vase
- scissors
- teddy bear
- hair drier
- toothbrush
86 changes: 86 additions & 0 deletions anylabeling/configs/auto_labeling/rtdetrv2m7x.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: rtdetrv2
name: rtdetrv2m7x-r20240730
display_name: RT-DETRv2-M* PaddleDetection
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.4.0/rtdetrv2_r50vd_m_7x_coco.onnx
score_threshold: 0.45
classes:
- person
- bicycle
- car
- motorcycle
- airplane
- bus
- train
- truck
- boat
- traffic light
- fire hydrant
- stop sign
- parking meter
- bench
- bird
- cat
- dog
- horse
- sheep
- cow
- elephant
- bear
- zebra
- giraffe
- backpack
- umbrella
- handbag
- tie
- suitcase
- frisbee
- skis
- snowboard
- sports ball
- kite
- baseball bat
- baseball glove
- skateboard
- surfboard
- tennis racket
- bottle
- wine glass
- cup
- fork
- knife
- spoon
- bowl
- banana
- apple
- sandwich
- orange
- broccoli
- carrot
- hot dog
- pizza
- donut
- cake
- chair
- couch
- potted plant
- bed
- dining table
- toilet
- tv
- laptop
- mouse
- remote
- keyboard
- cell phone
- microwave
- oven
- toaster
- sink
- refrigerator
- book
- clock
- vase
- scissors
- teddy bear
- hair drier
- toothbrush
86 changes: 86 additions & 0 deletions anylabeling/configs/auto_labeling/rtdetrv2s.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: rtdetrv2
name: rtdetrv2s-r20240730
display_name: RT-DETRv2-S PaddleDetection
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.4.0/rtdetrv2_r18vd_120e_coco.onnx
score_threshold: 0.45
classes:
- person
- bicycle
- car
- motorcycle
- airplane
- bus
- train
- truck
- boat
- traffic light
- fire hydrant
- stop sign
- parking meter
- bench
- bird
- cat
- dog
- horse
- sheep
- cow
- elephant
- bear
- zebra
- giraffe
- backpack
- umbrella
- handbag
- tie
- suitcase
- frisbee
- skis
- snowboard
- sports ball
- kite
- baseball bat
- baseball glove
- skateboard
- surfboard
- tennis racket
- bottle
- wine glass
- cup
- fork
- knife
- spoon
- bowl
- banana
- apple
- sandwich
- orange
- broccoli
- carrot
- hot dog
- pizza
- donut
- cake
- chair
- couch
- potted plant
- bed
- dining table
- toilet
- tv
- laptop
- mouse
- remote
- keyboard
- cell phone
- microwave
- oven
- toaster
- sink
- refrigerator
- book
- clock
- vase
- scissors
- teddy bear
- hair drier
- toothbrush
Loading

0 comments on commit 0fe5cff

Please sign in to comment.