Hugging Face's logo

infinity1096
/
UFM-Refine-980

Other
Safetensors
English
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
 Community
infinity1096 commited on
Commit
b120ac5
·
verified ·
1 Parent(s): 420c190

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +10 -0
  2. config.json +166 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - model_hub_mixin
4
+ - pytorch_model_hub_mixin
5
+ ---
6
+
7
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
+ - Code: [More Information Needed]
9
+ - Paper: [More Information Needed]
10
+ - Docs: [More Information Needed]
config.json ADDED
@@ -0,0 +1,166 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "adaptors_kwargs": {
3
+ "flow": {
4
+ "class": "FlowAdaptor",
5
+ "kwargs": {
6
+ "base_shape": [
7
+ 224,
8
+ 224
9
+ ],
10
+ "flow_mean": [
11
+ 0.0,
12
+ 0.0
13
+ ],
14
+ "flow_std": [
15
+ 25,
16
+ 25
17
+ ],
18
+ "name": "flow",
19
+ "output_normalized_coordinate": false,
20
+ "scale_strategy": "scale_both"
21
+ }
22
+ }
23
+ },
24
+ "classification_head_kwargs": {
25
+ "bias": true,
26
+ "input_feature_dim": "1024 + 768",
27
+ "mlp_ratio": 2,
28
+ "output_dim": 16,
29
+ "patch_size": 14,
30
+ "pretrained_checkpoint_path": null
31
+ },
32
+ "classification_head_type": "patch_mlp",
33
+ "detach_uncertainty_head": true,
34
+ "encoder_kwargs": {
35
+ "data_norm_type": "dinov2",
36
+ "indices": [
37
+ 6,
38
+ 23
39
+ ],
40
+ "name": "dinov2_encoder",
41
+ "patch_size": 14,
42
+ "size": "large",
43
+ "with_registers": false
44
+ },
45
+ "encoder_str": "dinov2",
46
+ "feature_combine_method": "modulate",
47
+ "feature_head_kwargs": {
48
+ "dpt_feature": {
49
+ "feature_dim": 256,
50
+ "hooks": [
51
+ 0,
52
+ 1,
53
+ 2,
54
+ 3
55
+ ],
56
+ "input_feature_dims": [
57
+ 1024,
58
+ 768,
59
+ 768,
60
+ 768
61
+ ],
62
+ "layer_dims": [
63
+ 96,
64
+ 192,
65
+ 384,
66
+ 768
67
+ ],
68
+ "output_width_ratio": 1,
69
+ "patch_size": 14,
70
+ "pretrained_checkpoint_path": null,
71
+ "use_bn": false
72
+ },
73
+ "dpt_processor": {
74
+ "hidden_dims": [
75
+ 128,
76
+ 128
77
+ ],
78
+ "input_feature_dim": 256,
79
+ "output_dim": 2,
80
+ "pretrained_checkpoint_path": null
81
+ }
82
+ },
83
+ "head_type": "dpt",
84
+ "inference_resolution": [
85
+ 560,
86
+ 420
87
+ ],
88
+ "info_sharing_and_head_structure": "dual+single",
89
+ "info_sharing_kwargs": {
90
+ "attn_drop": 0.0,
91
+ "custom_positional_encoding": null,
92
+ "depth": 12,
93
+ "dim": 768,
94
+ "drop_path": 0.0,
95
+ "gradient_checkpointing": false,
96
+ "indices": [
97
+ 5,
98
+ 8
99
+ ],
100
+ "init_values": null,
101
+ "input_embed_dim": 1024,
102
+ "max_num_views": 2,
103
+ "mlp_ratio": 4.0,
104
+ "name": "global_attention",
105
+ "norm_intermediate": true,
106
+ "num_heads": 12,
107
+ "pretrained_checkpoint_path": null,
108
+ "proj_drop": 0.0,
109
+ "qk_norm": false,
110
+ "qkv_bias": true,
111
+ "size": "base",
112
+ "use_rand_idx_pe_for_non_reference_views": false
113
+ },
114
+ "info_sharing_str": "global_attention",
115
+ "pretrained_backbone_checkpoint_path": null,
116
+ "pretrained_checkpoint_path": null,
117
+ "refinement_range": 5,
118
+ "temperature": 4,
119
+ "uncertainty_adaptors_kwargs": {
120
+ "non_occluded_mask": {
121
+ "class": "MaskAdaptor",
122
+ "kwargs": {
123
+ "name": "non_occluded_mask"
124
+ }
125
+ }
126
+ },
127
+ "uncertainty_head_kwargs": {
128
+ "dpt_feature": {
129
+ "feature_dim": 256,
130
+ "hooks": [
131
+ 0,
132
+ 1,
133
+ 2,
134
+ 3
135
+ ],
136
+ "input_feature_dims": [
137
+ 1024,
138
+ 768,
139
+ 768,
140
+ 768
141
+ ],
142
+ "layer_dims": [
143
+ 96,
144
+ 192,
145
+ 384,
146
+ 768
147
+ ],
148
+ "output_width_ratio": 1,
149
+ "patch_size": 14,
150
+ "pretrained_checkpoint_path": null,
151
+ "use_bn": false
152
+ },
153
+ "dpt_processor": {
154
+ "hidden_dims": [
155
+ 128,
156
+ 128
157
+ ],
158
+ "input_feature_dim": 256,
159
+ "output_dim": 1,
160
+ "pretrained_checkpoint_path": null
161
+ }
162
+ },
163
+ "uncertainty_head_type": "dpt",
164
+ "use_unet_batchnorm": false,
165
+ "use_unet_feature": true
166
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a85616fc4218cb19026702ec03fee0add4bc532a31ea7b5ebb5f2af756fb852e
3
+ size 1911126288