Hugging Face's logo

miroslavjaros
/
fruitsam

TensorBoard
Model card Files Metrics Community
miroslavjaros commited on
Commit
53622b5
•
1 Parent(s): abf6463

Pridani modelu SAMfruit

Browse files
Files changed (14) hide show
  1. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/assets.json +1 -0
  2. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/config.yaml +122 -0
  3. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/data_processors.pkl +3 -0
  4. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/df_preprocessor.pkl +3 -0
  5. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/events.out.tfevents.1704195475.user-ithope.6037.0 +3 -0
  6. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/hparams.yaml +15 -0
  7. 966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/model.ckpt +3 -0
  8. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/assets.json +1 -0
  9. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/config.yaml +122 -0
  10. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/data_processors.pkl +3 -0
  11. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/df_preprocessor.pkl +3 -0
  12. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/events.out.tfevents.1704197955.user-ithope.6037.2 +3 -0
  13. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/hparams.yaml +15 -0
  14. f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/model.ckpt +3 -0
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/assets.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"class_name": "SemanticSegmentationLearner", "column_types": {"image": "semantic_segmentation_img", "mask": "semantic_segmentation_gt"}, "label_column": "mask", "problem_type": "semantic_segmentation", "presets": null, "eval_metric_name": "iou", "validation_metric_name": "iou", "minmax_mode": "max", "output_shape": 1, "save_path": "/home/miro/Documents/colab/tmp/966dce5f981645f397cbfc80a121f57d-automm_semantic_seg", "pretrained": true, "pretrained_path": null, "fit_called": true, "best_score": null, "total_train_time": null, "version": "1.0.0"}
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/config.yaml ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ names:
3
+ - sam
4
+ sam:
5
+ checkpoint_name: facebook/sam-vit-base
6
+ image_norm: imagenet
7
+ data_types:
8
+ - semantic_segmentation_img
9
+ train_transforms:
10
+ - random_horizontal_flip
11
+ val_transforms: []
12
+ img_transforms:
13
+ - resize_to_square
14
+ gt_transforms:
15
+ - resize_gt_to_square
16
+ max_img_num_per_col: 1
17
+ frozen_layers:
18
+ - mask_decoder.iou_prediction_head
19
+ - prompt_encoder
20
+ num_mask_tokens: 1
21
+ ignore_label: 255
22
+ data:
23
+ image:
24
+ missing_value_strategy: zero
25
+ text:
26
+ normalize_text: false
27
+ categorical:
28
+ minimum_cat_count: 100
29
+ maximum_num_cat: 20
30
+ convert_to_text: true
31
+ numerical:
32
+ convert_to_text: false
33
+ scaler_with_mean: true
34
+ scaler_with_std: true
35
+ document:
36
+ missing_value_strategy: zero
37
+ label:
38
+ numerical_label_preprocessing: standardscaler
39
+ pos_label: null
40
+ column_features_pooling_mode: concat
41
+ mixup:
42
+ turn_on: false
43
+ mixup_alpha: 0.8
44
+ cutmix_alpha: 1.0
45
+ cutmix_minmax: null
46
+ prob: 1.0
47
+ switch_prob: 0.5
48
+ mode: batch
49
+ turn_off_epoch: 5
50
+ label_smoothing: 0.1
51
+ templates:
52
+ turn_on: false
53
+ num_templates: 30
54
+ template_length: 2048
55
+ preset_templates:
56
+ - super_glue
57
+ - rte
58
+ custom_templates: null
59
+ optimization:
60
+ optim_type: adamw
61
+ learning_rate: 0.0001
62
+ weight_decay: 0.0001
63
+ lr_choice: single_stage
64
+ lr_decay: 0
65
+ lr_schedule: polynomial_decay
66
+ max_epochs: 30
67
+ max_steps: -1
68
+ warmup_steps: 0.0
69
+ end_lr: 0
70
+ lr_mult: 1
71
+ patience: 10
72
+ val_check_interval: 1.0
73
+ check_val_every_n_epoch: 1
74
+ skip_final_val: false
75
+ gradient_clip_val: 1
76
+ gradient_clip_algorithm: norm
77
+ track_grad_norm: -1
78
+ log_every_n_steps: 10
79
+ top_k: 3
80
+ top_k_average_method: best
81
+ efficient_finetune: lora
82
+ lora:
83
+ module_filter:
84
+ - .*vision_encoder.*attn
85
+ filter:
86
+ - q
87
+ - v
88
+ r: 3
89
+ alpha: 32
90
+ loss_function: structure_loss
91
+ focal_loss:
92
+ alpha: null
93
+ gamma: 2.0
94
+ reduction: mean
95
+ mask2former_loss:
96
+ loss_cross_entropy_weight: 10.0
97
+ loss_mask_weight: 5.0
98
+ loss_dice_weight: 5.0
99
+ extra_trainable_params:
100
+ - .*mask_decoder
101
+ env:
102
+ num_gpus: 1
103
+ num_nodes: 1
104
+ batch_size: 4
105
+ per_gpu_batch_size: 1
106
+ eval_batch_size_ratio: 1
107
+ per_gpu_batch_size_evaluation: null
108
+ precision: 16-mixed
109
+ num_workers: 4
110
+ num_workers_evaluation: 2
111
+ accelerator: auto
112
+ fast_dev_run: false
113
+ deterministic: false
114
+ auto_select_gpus: false
115
+ strategy: auto
116
+ deepspeed_allgather_size: 1000000000.0
117
+ deepspeed_allreduce_size: 1000000000.0
118
+ compile:
119
+ turn_on: false
120
+ mode: default
121
+ dynamic: true
122
+ backend: inductor
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/data_processors.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f51cc52c846b44ba4adb742543e669d5c0fe4b78f6cd1e16b9cbe404042d80e2
3
+ size 13276
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/df_preprocessor.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fc376f598308074894f20fb74d17c5d55ed3beb6ac31dbfd4ae117a1978d400
3
+ size 11978
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/events.out.tfevents.1704195475.user-ithope.6037.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:956f5164b31f00f707fad052960ae27975e832831ca394cf219c240992a764bd
3
+ size 51266
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/hparams.yaml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ optim_type: adamw
2
+ lr_choice: single_stage
3
+ lr_schedule: polynomial_decay
4
+ lr: 0.0001
5
+ lr_decay: 0
6
+ end_lr: 0
7
+ lr_mult: 1
8
+ weight_decay: 0.0001
9
+ warmup_steps: 0.0
10
+ validation_metric_name: iou
11
+ custom_metric_func: null
12
+ efficient_finetune: lora
13
+ mixup_off_epoch: 5
14
+ skip_final_val: false
15
+ track_grad_norm: -1
966dce5f981645f397cbfc80a121f57d-automm_semantic_seg/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5619b241b4a011a67a5b48c7f5bb7c0af787a6b781c3c534f3762120f6d24481
3
+ size 373790141
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/assets.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"class_name": "SemanticSegmentationLearner", "column_types": {"image": "semantic_segmentation_img", "mask": "semantic_segmentation_gt"}, "label_column": "mask", "problem_type": "semantic_segmentation", "presets": null, "eval_metric_name": "iou", "validation_metric_name": "iou", "minmax_mode": "max", "output_shape": 1, "save_path": "/home/miro/Documents/colab/tmp/f876dc373c53463caaaf4e448749f57f-automm_semantic_seg", "pretrained": true, "pretrained_path": null, "fit_called": true, "best_score": null, "total_train_time": null, "version": "1.0.0"}
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/config.yaml ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ names:
3
+ - sam
4
+ sam:
5
+ checkpoint_name: facebook/sam-vit-base
6
+ image_norm: imagenet
7
+ data_types:
8
+ - semantic_segmentation_img
9
+ train_transforms:
10
+ - random_horizontal_flip
11
+ val_transforms: []
12
+ img_transforms:
13
+ - resize_to_square
14
+ gt_transforms:
15
+ - resize_gt_to_square
16
+ max_img_num_per_col: 1
17
+ frozen_layers:
18
+ - mask_decoder.iou_prediction_head
19
+ - prompt_encoder
20
+ num_mask_tokens: 1
21
+ ignore_label: 255
22
+ data:
23
+ image:
24
+ missing_value_strategy: zero
25
+ text:
26
+ normalize_text: false
27
+ categorical:
28
+ minimum_cat_count: 100
29
+ maximum_num_cat: 20
30
+ convert_to_text: true
31
+ numerical:
32
+ convert_to_text: false
33
+ scaler_with_mean: true
34
+ scaler_with_std: true
35
+ document:
36
+ missing_value_strategy: zero
37
+ label:
38
+ numerical_label_preprocessing: standardscaler
39
+ pos_label: null
40
+ column_features_pooling_mode: concat
41
+ mixup:
42
+ turn_on: false
43
+ mixup_alpha: 0.8
44
+ cutmix_alpha: 1.0
45
+ cutmix_minmax: null
46
+ prob: 1.0
47
+ switch_prob: 0.5
48
+ mode: batch
49
+ turn_off_epoch: 5
50
+ label_smoothing: 0.1
51
+ templates:
52
+ turn_on: false
53
+ num_templates: 30
54
+ template_length: 2048
55
+ preset_templates:
56
+ - super_glue
57
+ - rte
58
+ custom_templates: null
59
+ optimization:
60
+ optim_type: adamw
61
+ learning_rate: 0.0001
62
+ weight_decay: 0.0001
63
+ lr_choice: single_stage
64
+ lr_decay: 0
65
+ lr_schedule: polynomial_decay
66
+ max_epochs: 30
67
+ max_steps: -1
68
+ warmup_steps: 0.0
69
+ end_lr: 0
70
+ lr_mult: 1
71
+ patience: 10
72
+ val_check_interval: 1.0
73
+ check_val_every_n_epoch: 1
74
+ skip_final_val: false
75
+ gradient_clip_val: 1
76
+ gradient_clip_algorithm: norm
77
+ track_grad_norm: -1
78
+ log_every_n_steps: 10
79
+ top_k: 3
80
+ top_k_average_method: best
81
+ efficient_finetune: lora
82
+ lora:
83
+ module_filter:
84
+ - .*vision_encoder.*attn
85
+ filter:
86
+ - q
87
+ - v
88
+ r: 3
89
+ alpha: 32
90
+ loss_function: structure_loss
91
+ focal_loss:
92
+ alpha: null
93
+ gamma: 2.0
94
+ reduction: mean
95
+ mask2former_loss:
96
+ loss_cross_entropy_weight: 10.0
97
+ loss_mask_weight: 5.0
98
+ loss_dice_weight: 5.0
99
+ extra_trainable_params:
100
+ - .*mask_decoder
101
+ env:
102
+ num_gpus: 1
103
+ num_nodes: 1
104
+ batch_size: 4
105
+ per_gpu_batch_size: 1
106
+ eval_batch_size_ratio: 1
107
+ per_gpu_batch_size_evaluation: null
108
+ precision: 16-mixed
109
+ num_workers: 4
110
+ num_workers_evaluation: 2
111
+ accelerator: auto
112
+ fast_dev_run: false
113
+ deterministic: false
114
+ auto_select_gpus: false
115
+ strategy: auto
116
+ deepspeed_allgather_size: 1000000000.0
117
+ deepspeed_allreduce_size: 1000000000.0
118
+ compile:
119
+ turn_on: false
120
+ mode: default
121
+ dynamic: true
122
+ backend: inductor
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/data_processors.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f51cc52c846b44ba4adb742543e669d5c0fe4b78f6cd1e16b9cbe404042d80e2
3
+ size 13276
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/df_preprocessor.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fc376f598308074894f20fb74d17c5d55ed3beb6ac31dbfd4ae117a1978d400
3
+ size 11978
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/events.out.tfevents.1704197955.user-ithope.6037.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77ce82298849be0d882d6509dbf454ada3eef182ec294bda0de75a0ca122386d
3
+ size 258770
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/hparams.yaml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ optim_type: adamw
2
+ lr_choice: single_stage
3
+ lr_schedule: polynomial_decay
4
+ lr: 0.0001
5
+ lr_decay: 0
6
+ end_lr: 0
7
+ lr_mult: 1
8
+ weight_decay: 0.0001
9
+ warmup_steps: 0.0
10
+ validation_metric_name: iou
11
+ custom_metric_func: null
12
+ efficient_finetune: lora
13
+ mixup_off_epoch: 5
14
+ skip_final_val: false
15
+ track_grad_norm: -1
f876dc373c53463caaaf4e448749f57f-automm_semantic_seg/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbef06149f8be6d52a512ebefd382a795ff40c7c4e7f1435fad979530d5b654b
3
+ size 373790141