Hugging Face's logo

davelop
/
face_seg

Image Segmentation
Model card Files Community
face_seg
File size: 865 Bytes 
25d497c
8aaf46b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0a32b99
8aaf46b
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
{
  "model_name": "CustomUNet",
  "description": "A custom U-Net model for semantic segmentation using TensorFlow/Keras.",
  "input_shape": [512, 512, 3],  // Input shape of the images
  "num_classes": 11,  // Number of output classes
  "architecture": {
    "encoder": {
      "type": "custom",
      "blocks": [
        {"filters": 64},
        {"filters": 128},
        {"filters": 256},
        {"filters": 512}
      ]
    },
    "bottleneck": {
      "filters": 1024
    },
    "decoder": {
      "type": "custom",
      "blocks": [
        {"filters": 512},
        {"filters": 256},
        {"filters": 128},
        {"filters": 64}
      ]
    }
  },
  "loss_function": "categorical_crossentropy",
  "optimizer": {
    "type": "Adam",
    "learning_rate": 0.001
  },
  "training": {
    "batch_size": 16,
    "epochs": 50,
    "validation_split": 0.2
  }
}