Skip to content

Commit

Permalink
[Enhancement] Support yolov10 model
Browse files Browse the repository at this point in the history
  • Loading branch information
CVHub520 committed May 25, 2024
1 parent dca185a commit b73c268
Show file tree
Hide file tree
Showing 16 changed files with 784 additions and 5 deletions.
2 changes: 1 addition & 1 deletion anylabeling/app_info.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
__appname__ = "X-AnyLabeling"
__appdescription__ = "Advanced Auto Labeling Solution with Added Features"
__version__ = "2.3.5"
__version__ = "2.3.6"
__preferred_device__ = "CPU" # GPU or CPU
12 changes: 12 additions & 0 deletions anylabeling/configs/auto_labeling/models.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,18 @@
config_file: ":/yolov9c.yaml"
- model_name: "yolov9e-r20240224"
config_file: ":/yolov9e.yaml"
- model_name: "yolov10b-r20240525"
config_file: ":/yolov10b.yaml"
- model_name: "yolov10l-r20240525"
config_file: ":/yolov10l.yaml"
- model_name: "yolov10m-r20240525"
config_file: ":/yolov10m.yaml"
- model_name: "yolov10n-r20240525"
config_file: ":/yolov10n.yaml"
- model_name: "yolov10s-r20240525"
config_file: ":/yolov10s.yaml"
- model_name: "yolov10x-r20240525"
config_file: ":/yolov10x.yaml"
- model_name: "yolow_l-r20240227"
config_file: ":/yolow_l.yaml"
- model_name: "yolox_l_dwpose_ucoco-r20230820"
Expand Down
86 changes: 86 additions & 0 deletions anylabeling/configs/auto_labeling/yolov10b.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: yolov10
name: yolov10b-r20240525
display_name: YOLOv10b THU-MIG
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.3.6/yolov10b.onnx
confidence_threshold: 0.25
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/yolov10l.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: yolov10
name: yolov10l-r20240525
display_name: YOLOv10l THU-MIG
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.3.6/yolov10l.onnx
confidence_threshold: 0.25
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/yolov10m.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: yolov10
name: yolov10m-r20240525
display_name: YOLOv10m THU-MIG
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.3.6/yolov10m.onnx
confidence_threshold: 0.25
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/yolov10n.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
type: yolov10
name: yolov10n-r20240525
display_name: YOLOv10n THU-MIG
model_path: https://github.com/CVHub520/X-AnyLabeling/releases/download/v2.3.6/yolov10n.onnx
confidence_threshold: 0.25
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 b73c268

Please sign in to comment.