all yolov5 models are supported

anylabeling/configs/autolabel_yolov5l.yaml

@@ -0,0 +1,90 @@
+type: yolov5
+name: yolov5l
+display_name: YOLOv5l Ultralytics
+model_path: anylabeling_assets/models/yolov5/yolov5l.onnx
+input_width: 640
+input_height: 640
+score_threshold: 0.5
+nms_threshold: 0.45
+confidence_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

anylabeling/configs/autolabel_yolov5n.yaml

@@ -0,0 +1,90 @@
+type: yolov5
+name: yolov5n
+display_name: YOLOv5n Ultralytics
+model_path: anylabeling_assets/models/yolov5/yolov5n.onnx
+input_width: 640
+input_height: 640
+score_threshold: 0.5
+nms_threshold: 0.45
+confidence_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

anylabeling/configs/autolabel_yolov5x.yaml

@@ -0,0 +1,90 @@
+type: yolov5
+name: yolov5x
+display_name: YOLOv5x Ultralytics
+model_path: anylabeling_assets/models/yolov5/yolov5x.onnx
+input_width: 640
+input_height: 640
+score_threshold: 0.5
+nms_threshold: 0.45
+confidence_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

anylabeling/services/auto_labeling/model.py

@@ -11,7 +11,7 @@
 
 class Model:
     BASE_DOWNLOAD_URL = (
-        "https://github.com/vietanhdev/anylabeling-assets/raw/main/"
+        "https://github.com/hdnh2006/anylabeling-assets/releases/download/v0.0.0/"
     )
 
     class Meta:
@@ -57,8 +57,8 @@ def get_model_abs_path(self, model_path):
                 " while..."
             )
             relative_path = model_path.replace("anylabeling_assets/", "")
-            download_url = self.BASE_DOWNLOAD_URL + relative_path
-            home_dir = os.path.expanduser("~")
+            download_url = self.BASE_DOWNLOAD_URL + relative_path.replace("models/yolov5/","")
+            home_dir = os.getcwd()
             model_abs_path = os.path.abspath(
                 os.path.join(home_dir, "data", relative_path)
             )

anylabeling/services/auto_labeling/model_manager.py

@@ -21,10 +21,14 @@ class ModelManager(QObject):
 
     model_configs = {
         "segment_anything_vit_b": "autolabel_segment_anything.yaml",
+        "yolov5n": "autolabel_yolov5n.yaml",
         "yolov5s": "autolabel_yolov5s.yaml",
         "yolov5m": "autolabel_yolov5m.yaml",
+        "yolov5l": "autolabel_yolov5l.yaml",
+        "yolov5x": "autolabel_yolov5x.yaml",
     }
 
+
     def __init__(self):
         super().__init__()
         self.model_infos = {}