Add model

This commit is contained in:
Niels Rogge 2022-02-24 16:11:32 +01:00
parent 837a503bc0
commit c8d581da9d
2 changed files with 40 additions and 0 deletions

37
config.json Normal file
View File

@ -0,0 +1,37 @@
{
"architectures": [
"DPTForDepthEstimation"
],
"attention_probs_dropout_prob": 0.0,
"channels": 256,
"expand_channels": false,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"image_size": 384,
"initializer_range": 0.02,
"intermediate_size": 4096,
"layer_norm_eps": 1e-12,
"model_type": "dpt",
"num_attention_heads": 16,
"num_channels": 3,
"num_hidden_layers": 24,
"out_indices": [
5,
11,
17,
23
],
"patch_size": 16,
"post_process_channels": [
256,
512,
1024,
1024
],
"qkv_bias": true,
"readout_type": "project",
"torch_dtype": "float32",
"transformers_version": "4.17.0.dev0",
"use_batch_norm": false
}

3
pytorch_model.bin Normal file
View File

@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:71bcb66eb63300ff02109eadaf311fb6c3314e7a10bd3a50d5d9ccfe808a4a6b
size 1367570865