diff --git a/config.json b/config.json new file mode 100644 index 0000000..d46def8 --- /dev/null +++ b/config.json @@ -0,0 +1,47 @@ +{ + "architectures": [ + "DPTForDepthEstimation" + ], + "attention_probs_dropout_prob": 0.1, + "auxiliary_loss_weight": 0.4, + "backbone_out_indices": [ + 0, + 1, + 2, + 3 + ], + "fusion_hidden_size": 256, + "head_in_index": -1, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.1, + "hidden_size": 32, + "image_size": 32, + "initializer_range": 0.02, + "intermediate_size": 37, + "layer_norm_eps": 1e-12, + "model_type": "dpt", + "neck_hidden_sizes": [ + 96, + 192, + 384, + 768 + ], + "num_attention_heads": 4, + "num_channels": 3, + "num_hidden_layers": 4, + "patch_size": 16, + "qkv_bias": true, + "readout_type": "project", + "reassemble_factors": [ + 4, + 2, + 1, + 0.5 + ], + "semantic_classifier_dropout": 0.1, + "semantic_loss_ignore_index": 255, + "torch_dtype": "float32", + "transformers_version": "4.25.0.dev0", + "use_auxiliary_head": true, + "use_batch_norm_in_fusion_residual": false +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..6b8d1e6 --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,25 @@ +{ + "crop_size": 32, + "do_normalize": true, + "do_rescale": true, + "do_resize": true, + "ensure_multiple_of": 1, + "image_mean": [ + 0.5, + 0.5, + 0.5 + ], + "image_processor_type": "DPTImageProcessor", + "image_std": [ + 0.5, + 0.5, + 0.5 + ], + "keep_aspect_ratio": false, + "resample": 2, + "rescale_factor": 0.00392156862745098, + "size": { + "height": 32, + "width": 32 + } +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..11e379c --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:746a123575ccaf12c050845f42b2a76b1fdac47ab1d1c0da6afa4573dab0916a +size 76312069