From dafba0331075ccde33da85f08ad2767bef7cc5b6 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 15 Oct 2021 09:49:59 +0000 Subject: [PATCH] First commit --- config.json | 110 +++++++++++++++++++++++++++++++++++++++ preprocessor_config.json | 32 ++++++++++++ pytorch_model.bin | 3 ++ 3 files changed, 145 insertions(+) create mode 100644 config.json create mode 100644 preprocessor_config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..66815fd --- /dev/null +++ b/config.json @@ -0,0 +1,110 @@ +{ + "architectures": [ + "SegformerForImageSegmentation" + ], + "attention_probs_dropout_prob": 0.0, + "classifier_dropout_prob": 0.1, + "decoder_hidden_size": 768, + "depths": [ + 3, + 6, + 40, + 3 + ], + "downsampling_rates": [ + 1, + 4, + 8, + 16 + ], + "drop_path_rate": 0.1, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.0, + "hidden_sizes": [ + 64, + 128, + 320, + 512 + ], + "id2label": { + "0": "road", + "1": "sidewalk", + "2": "building", + "3": "wall", + "4": "fence", + "5": "pole", + "6": "traffic light", + "7": "traffic sign", + "8": "vegetation", + "9": "terrain", + "10": "sky", + "11": "person", + "12": "rider", + "13": "car", + "14": "truck", + "15": "bus", + "16": "train", + "17": "motorcycle", + "18": "bicycle" + }, + "image_size": 224, + "initializer_range": 0.02, + "label2id": { + "bicycle": 18, + "building": 2, + "bus": 15, + "car": 13, + "fence": 4, + "motorcycle": 17, + "person": 11, + "pole": 5, + "rider": 12, + "road": 0, + "sidewalk": 1, + "sky": 10, + "terrain": 9, + "traffic light": 6, + "traffic sign": 7, + "train": 16, + "truck": 14, + "vegetation": 8, + "wall": 3 + }, + "layer_norm_eps": 1e-06, + "mlp_ratios": [ + 4, + 4, + 4, + 4 + ], + "model_type": "segformer", + "num_attention_heads": [ + 1, + 2, + 5, + 8 + ], + "num_channels": 3, + "num_encoder_blocks": 4, + "patch_sizes": [ + 7, + 3, + 3, + 3 + ], + "reshape_last_stage": true, + "sr_ratios": [ + 8, + 4, + 2, + 1 + ], + "strides": [ + 4, + 2, + 2, + 2 + ], + "torch_dtype": "float32", + "transformers_version": "4.12.0.dev0" +} diff --git a/preprocessor_config.json b/preprocessor_config.json new file mode 100644 index 0000000..f9ec06e --- /dev/null +++ b/preprocessor_config.json @@ -0,0 +1,32 @@ +{ + "align": false, + "crop_size": [ + 512, + 512 + ], + "do_normalize": true, + "do_pad": true, + "do_random_crop": false, + "do_resize": true, + "feature_extractor_type": "SegformerFeatureExtractor", + "image_mean": [ + 0.485, + 0.456, + 0.406 + ], + "image_scale": [ + 512, + 512 + ], + "image_std": [ + 0.229, + 0.224, + 0.225 + ], + "keep_ratio": false, + "padding_value": 0, + "reduce_zero_label": false, + "resample": 2, + "segmentation_padding_value": -100, + "size_divisor": 32 +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..ba0798a --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bdefc7b5bf4b548d347e6bcbe3ab1a0ff4370047211ae9e56c9e53dfb1f78033 +size 338896453