Hugging Face's logo

Saving-Willy
/
cetacean-classifier

Image Classification
Transformers
Safetensors
cetaceanet
biology
biodiversity
custom_code
Model card Files Community
MalloryWittwerEPFL commited on
Commit
346dd48
·
verified ·
1 Parent(s): b3201aa

Upload model

Browse files
Files changed (2) hide show
  1. config.json +85 -85
  2. model.safetensors +1 -1
config.json CHANGED
@@ -1,85 +1,85 @@
1
- {
2
- "architectures": [
3
- "CetaceanClassifierModelForImageClassification"
4
- ],
5
- "aug": {
6
- "crop_l": 0.75,
7
- "crop_r": 1.3333333333333333,
8
- "crop_scale": 0.9,
9
- "p_affine": 0.5,
10
- "p_blur": 0.05,
11
- "p_bright_contrast": 0.5,
12
- "p_cutout": 0.05,
13
- "p_downscale": 0.0,
14
- "p_gray": 0.1,
15
- "p_noise": 0.05,
16
- "p_posterize": 0.2,
17
- "p_rain": 0.05,
18
- "p_shuffle": 0.3,
19
- "p_snow": 0.1,
20
- "rotate": 15,
21
- "shear": 3,
22
- "translate": 0.25
23
- },
24
- "auto_map": {
25
- "AutoConfig": "configuration_cetacean_classifier.CetaceanClassifierConfig",
26
- "AutoModelForImageClassification": "modeling_cetacean_classifier.CetaceanClassifierModelForImageClassification"
27
- },
28
- "batch_size": 8,
29
- "bbox_conf_threshold": 0.01,
30
- "bboxes": {
31
- "backfin": 0.15,
32
- "detic": 0.05,
33
- "fullbody": 0.6,
34
- "fullbody_charm": 0.15,
35
- "none": 0.05
36
- },
37
- "global_pool": {
38
- "arch": "GeM",
39
- "p": 3,
40
- "train": false
41
- },
42
- "image_size": [
43
- 768,
44
- 768
45
- ],
46
- "loss_fn": "CrossEntropy",
47
- "loss_id_ratio": 0.437338,
48
- "lr_backbone": 0.0016,
49
- "lr_decay_scale": 0.01,
50
- "lr_head": 0.016,
51
- "margin_coef_id": 0.27126,
52
- "margin_coef_species": 0.226253,
53
- "margin_cons_id": 0.05,
54
- "margin_cons_species": 0.05,
55
- "margin_power_id": -0.364399,
56
- "margin_power_species": -0.720133,
57
- "max_epochs": 30,
58
- "model_name": "tf_efficientnetv2_m",
59
- "model_type": "cetaceanet",
60
- "n_center_id": 2,
61
- "n_center_species": 2,
62
- "n_data": -1,
63
- "n_splits": -1,
64
- "normalization": "batchnorm",
65
- "num_classes": 15587,
66
- "num_species_classes": 26,
67
- "optimizer": "AdamW",
68
- "out_indices": [
69
- 3,
70
- 4
71
- ],
72
- "pretrained": true,
73
- "pseudo_conf_threshold": 0.6,
74
- "pseudo_label": "pseudo_labels/round2.csv",
75
- "s_id": 20.9588,
76
- "s_species": 33.1383,
77
- "test_bboxes": [
78
- "fullbody",
79
- "fullbody_charm"
80
- ],
81
- "torch_dtype": "float32",
82
- "transformers_version": "4.46.0",
83
- "val_bbox": "fullbody",
84
- "warmup_steps_ratio": 0.2
85
- }
 
1
+ {
2
+ "architectures": [
3
+ "CetaceanClassifierModelForImageClassification"
4
+ ],
5
+ "aug": {
6
+ "crop_l": 0.75,
7
+ "crop_r": 1.3333333333333333,
8
+ "crop_scale": 0.9,
9
+ "p_affine": 0.5,
10
+ "p_blur": 0.05,
11
+ "p_bright_contrast": 0.5,
12
+ "p_cutout": 0.05,
13
+ "p_downscale": 0.0,
14
+ "p_gray": 0.1,
15
+ "p_noise": 0.05,
16
+ "p_posterize": 0.2,
17
+ "p_rain": 0.05,
18
+ "p_shuffle": 0.3,
19
+ "p_snow": 0.1,
20
+ "rotate": 15,
21
+ "shear": 3,
22
+ "translate": 0.25
23
+ },
24
+ "auto_map": {
25
+ "AutoConfig": "configuration_cetacean_classifier.CetaceanClassifierConfig",
26
+ "AutoModelForImageClassification": "modeling_cetacean_classifier.CetaceanClassifierModelForImageClassification"
27
+ },
28
+ "batch_size": 8,
29
+ "bbox_conf_threshold": 0.01,
30
+ "bboxes": {
31
+ "backfin": 0.15,
32
+ "detic": 0.05,
33
+ "fullbody": 0.6,
34
+ "fullbody_charm": 0.15,
35
+ "none": 0.05
36
+ },
37
+ "global_pool": {
38
+ "arch": "GeM",
39
+ "p": 3,
40
+ "train": false
41
+ },
42
+ "image_size": [
43
+ 768,
44
+ 768
45
+ ],
46
+ "loss_fn": "CrossEntropy",
47
+ "loss_id_ratio": 0.437338,
48
+ "lr_backbone": 0.0016,
49
+ "lr_decay_scale": 0.01,
50
+ "lr_head": 0.016,
51
+ "margin_coef_id": 0.27126,
52
+ "margin_coef_species": 0.226253,
53
+ "margin_cons_id": 0.05,
54
+ "margin_cons_species": 0.05,
55
+ "margin_power_id": -0.364399,
56
+ "margin_power_species": -0.720133,
57
+ "max_epochs": 30,
58
+ "model_name": "tf_efficientnetv2_m",
59
+ "model_type": "cetaceanet",
60
+ "n_center_id": 2,
61
+ "n_center_species": 2,
62
+ "n_data": -1,
63
+ "n_splits": -1,
64
+ "normalization": "batchnorm",
65
+ "num_classes": 15587,
66
+ "num_species_classes": 26,
67
+ "optimizer": "AdamW",
68
+ "out_indices": [
69
+ 3,
70
+ 4
71
+ ],
72
+ "pretrained": true,
73
+ "pseudo_conf_threshold": 0.6,
74
+ "pseudo_label": "pseudo_labels/round2.csv",
75
+ "s_id": 20.9588,
76
+ "s_species": 33.1383,
77
+ "test_bboxes": [
78
+ "fullbody",
79
+ "fullbody_charm"
80
+ ],
81
+ "torch_dtype": "float32",
82
+ "transformers_version": "4.46.0",
83
+ "val_bbox": "fullbody",
84
+ "warmup_steps_ratio": 0.2
85
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:faf081f38c0224790f5819d763475e9827667fa50af1a86a61e233c52876ae11
3
  size 296028464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd07eaf5ca7a871b9162257e44e9bade9312ef1378ea1d77476b35253dda14dd
3
  size 296028464