justinsoberano commited on
Commit
0027ce9
1 Parent(s): fee29fb

Uploaded trained depth-prediction model

Browse files
config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "justinsoberano/depth-prediction-ai",
3
+ "architectures": [
4
+ "GLPNForDepthEstimation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 64,
9
+ "depths": [
10
+ 3,
11
+ 8,
12
+ 27,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "head_in_index": -1,
23
+ "hidden_act": "gelu",
24
+ "hidden_dropout_prob": 0.0,
25
+ "hidden_sizes": [
26
+ 64,
27
+ 128,
28
+ 320,
29
+ 512
30
+ ],
31
+ "image_size": 224,
32
+ "initializer_range": 0.02,
33
+ "layer_norm_eps": 1e-06,
34
+ "max_depth": 1,
35
+ "mlp_ratios": [
36
+ 4,
37
+ 4,
38
+ 4,
39
+ 4
40
+ ],
41
+ "model_type": "glpn",
42
+ "num_attention_heads": [
43
+ 1,
44
+ 2,
45
+ 5,
46
+ 8
47
+ ],
48
+ "num_channels": 3,
49
+ "num_encoder_blocks": 4,
50
+ "patch_sizes": [
51
+ 7,
52
+ 3,
53
+ 3,
54
+ 3
55
+ ],
56
+ "sr_ratios": [
57
+ 8,
58
+ 4,
59
+ 2,
60
+ 1
61
+ ],
62
+ "strides": [
63
+ 4,
64
+ 2,
65
+ 2,
66
+ 2
67
+ ],
68
+ "torch_dtype": "float32",
69
+ "transformers_version": "4.24.0"
70
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_rescale": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "GLPNImageProcessor",
5
+ "resample": 2,
6
+ "size_divisor": 32
7
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48b7377b0343e784c748734f9c035555a83ebf0f4f9099595c0fd0510ffc50a2
3
+ size 245212393
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fa52d5a1a46aefc7a522f3b10e99c7d6da808e029654f70c8e77168aa0ba6a5
3
+ size 3439