face_and_mobile_detection / preprocessor_config.json
sreenathsree1578's picture
Initial model upload
ed8a319
{
"image_size": [
224,
224
],
"batch_size": 32,
"num_classes": 2,
"class_names": [
"face",
"mobile_phone"
],
"train_split": 0.8,
"val_split": 0.1,
"test_split": 0.1,
"augmentation": {
"horizontal_flip": true,
"rotation_range": 10,
"width_shift_range": 0.1,
"height_shift_range": 0.1,
"zoom_range": 0.1,
"brightness_range": [
0.8,
1.2
]
},
"normalization": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
}
}