Skip to content

Commit 1220381

Browse files
vietanhdevvietanhdev
authored
Merge pull request #26 from vietanhdev/feature/add_segment_anything_models
Feature: Add all segment anything models
2 parents cfac668 + e6c2670 commit 1220381

21 files changed

Lines changed: 105 additions & 21 deletions

‎anylabeling.spec‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ a = Analysis(
1010
pathex=['anylabeling'],
1111
binaries=[],
1212
datas=[
13+
('anylabeling/configs/auto_labeling/*.yaml', 'anylabeling/configs/auto_labeling'),
1314
('anylabeling/configs/*.yaml', 'anylabeling/configs'),
1415
('anylabeling/views/labeling/icons/*', 'anylabeling/views/labeling/icons'),
1516
('anylabeling/resources/images/*', 'anylabeling/resources/images'),

‎anylabeling/configs/auto_labeling/__init__.py‎

Whitespace-only changes.

‎anylabeling/configs/auto_labeling/models.yaml‎

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
- model_name: "segment_anything_vit_b_quant-r20230416"
2+
config_file: "segment_anything_vit_b_quant.yaml"
3+
- model_name: "segment_anything_vit_b-r20230416"
4+
config_file: "segment_anything_vit_b.yaml"
5+
- model_name: "segment_anything_vit_l_quant-r20230416"
6+
config_file: "segment_anything_vit_l_quant.yaml"
7+
- model_name: "segment_anything_vit_l-r20230416"
8+
config_file: "segment_anything_vit_l.yaml"
9+
- model_name: "segment_anything_vit_h_quant-r20230416"
10+
config_file: "segment_anything_vit_h_quant.yaml"
11+
- model_name: "yolov5n-r20230415"
12+
config_file: "yolov5n.yaml"
13+
- model_name: "yolov5s-r20230415"
14+
config_file: "yolov5s.yaml"
15+
- model_name: "yolov5m-r20230415"
16+
config_file: "yolov5m.yaml"
17+
- model_name: "yolov5l-r20230415"
18+
config_file: "yolov5l.yaml"
19+
- model_name: "yolov5x-r20230415"
20+
config_file: "yolov5x.yaml"
21+
- model_name: "yolov8n-r20230415"
22+
config_file: "yolov8n.yaml"
23+
- model_name: "yolov8s-r20230415"
24+
config_file: "yolov8s.yaml"
25+
- model_name: "yolov8m-r20230415"
26+
config_file: "yolov8m.yaml"
27+
- model_name: "yolov8l-r20230415"
28+
config_file: "yolov8l.yaml"
29+
- model_name: "yolov8x-r20230415"
30+
config_file: "yolov8x.yaml"

anylabeling/configs/autolabel_segment_anything.yaml renamed to anylabeling/configs/auto_labeling/segment_anything_vit_b.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
type: segment_anything
2-
name: segment_anything_vit_b-r20230415
2+
name: segment_anything_vit_b-r20230416
33
display_name: Segment Anything (ViT-B)
4-
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.0.1/vit_b-encoder-quant.onnx
5-
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.0.1/vit_b-decoder-quant.onnx
4+
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_b_encoder.onnx
5+
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_b_decoder.onnx
66
input_size: 1024
77
max_width: 1024
88
max_height: 682

‎anylabeling/configs/auto_labeling/segment_anything_vit_b_quant.yaml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
type: segment_anything
2+
name: segment_anything_vit_b_quant-r20230416
3+
display_name: Segment Anything (ViT-B Quant)
4+
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_b_encoder_quant.onnx
5+
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_b_decoder_quant.onnx
6+
input_size: 1024
7+
max_width: 1024
8+
max_height: 682

‎anylabeling/configs/auto_labeling/segment_anything_vit_h_quant.yaml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
type: segment_anything
2+
name: segment_anything_vit_h_quant-r20230416
3+
display_name: Segment Anything (ViT-H Quant)
4+
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_h_encoder_quant.onnx
5+
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_h_decoder_quant.onnx
6+
input_size: 1024
7+
max_width: 1024
8+
max_height: 682

‎anylabeling/configs/auto_labeling/segment_anything_vit_l.yaml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
type: segment_anything
2+
name: segment_anything_vit_l-r20230416
3+
display_name: Segment Anything (ViT-L)
4+
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_l_encoder.onnx
5+
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_l_decoder.onnx
6+
input_size: 1024
7+
max_width: 1024
8+
max_height: 682

‎anylabeling/configs/auto_labeling/segment_anything_vit_l_quant.yaml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
type: segment_anything
2+
name: segment_anything_vit_l_quant-r20230416
3+
display_name: Segment Anything (ViT-L Quant)
4+
encoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_l_encoder_quant.onnx
5+
decoder_model_path: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.2.0/segment_anything_vit_l_decoder_quant.onnx
6+
input_size: 1024
7+
max_width: 1024
8+
max_height: 682

anylabeling/configs/autolabel_yolov5l.yaml renamed to anylabeling/configs/auto_labeling/yolov5l.yaml

File renamed without changes.

anylabeling/configs/autolabel_yolov5m.yaml renamed to anylabeling/configs/auto_labeling/yolov5m.yaml

File renamed without changes.

0 commit comments

Comments
 (0)