Upload 3 files
Browse files- model_config.json +59 -0
- phone_detection_model.h5 +3 -0
- preprocessor_config(1).json +59 -0
model_config.json
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "phone_detection_cnn",
|
| 3 |
+
"architecture": "MobileNetV2_with_custom_head",
|
| 4 |
+
"input_shape": [
|
| 5 |
+
224,
|
| 6 |
+
224,
|
| 7 |
+
3
|
| 8 |
+
],
|
| 9 |
+
"num_classes": 1,
|
| 10 |
+
"task": "binary_classification",
|
| 11 |
+
"classes": {
|
| 12 |
+
"0": "negative",
|
| 13 |
+
"1": "positive"
|
| 14 |
+
},
|
| 15 |
+
"training_parameters": {
|
| 16 |
+
"batch_size": 32,
|
| 17 |
+
"learning_rate": 0.001,
|
| 18 |
+
"optimizer": "Adam",
|
| 19 |
+
"loss_function": "binary_crossentropy",
|
| 20 |
+
"metrics": [
|
| 21 |
+
"accuracy"
|
| 22 |
+
],
|
| 23 |
+
"epochs": 50
|
| 24 |
+
},
|
| 25 |
+
"data_augmentation": {
|
| 26 |
+
"rotation_range": 20,
|
| 27 |
+
"width_shift_range": 0.2,
|
| 28 |
+
"height_shift_range": 0.2,
|
| 29 |
+
"horizontal_flip": true,
|
| 30 |
+
"zoom_range": 0.2,
|
| 31 |
+
"shear_range": 0.2,
|
| 32 |
+
"fill_mode": "nearest"
|
| 33 |
+
},
|
| 34 |
+
"model_layers": {
|
| 35 |
+
"base_model": "MobileNetV2",
|
| 36 |
+
"base_model_trainable": false,
|
| 37 |
+
"custom_layers": [
|
| 38 |
+
"GlobalAveragePooling2D",
|
| 39 |
+
"Dense_128_relu",
|
| 40 |
+
"Dropout_0.3",
|
| 41 |
+
"Dense_64_relu",
|
| 42 |
+
"Dropout_0.2",
|
| 43 |
+
"Dense_1_sigmoid"
|
| 44 |
+
]
|
| 45 |
+
},
|
| 46 |
+
"performance": {
|
| 47 |
+
"validation_accuracy": 0.8079096078872681,
|
| 48 |
+
"validation_loss": 0.4199891686439514
|
| 49 |
+
},
|
| 50 |
+
"dataset_info": {
|
| 51 |
+
"total_images": 888,
|
| 52 |
+
"training_samples": 711,
|
| 53 |
+
"validation_samples": 177,
|
| 54 |
+
"class_distribution": {
|
| 55 |
+
"positive": 540,
|
| 56 |
+
"negative": 348
|
| 57 |
+
}
|
| 58 |
+
}
|
| 59 |
+
}
|
phone_detection_model.h5
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:11d65a69ed7ca5553cea5976171a9939931c49b02a1abb6eb1296a6ef8887f54
|
| 3 |
+
size 11479248
|
preprocessor_config(1).json
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"preprocessing": {
|
| 3 |
+
"image_size": [
|
| 4 |
+
224,
|
| 5 |
+
224
|
| 6 |
+
],
|
| 7 |
+
"color_mode": "rgb",
|
| 8 |
+
"normalization": {
|
| 9 |
+
"rescale": 0.00392156862745098,
|
| 10 |
+
"description": "Pixel values normalized to [0,1]"
|
| 11 |
+
},
|
| 12 |
+
"data_format": "channels_last"
|
| 13 |
+
},
|
| 14 |
+
"data_pipeline": {
|
| 15 |
+
"training_augmentation": {
|
| 16 |
+
"rotation_range": 20,
|
| 17 |
+
"width_shift_range": 0.2,
|
| 18 |
+
"height_shift_range": 0.2,
|
| 19 |
+
"horizontal_flip": true,
|
| 20 |
+
"zoom_range": 0.2,
|
| 21 |
+
"shear_range": 0.2,
|
| 22 |
+
"fill_mode": "nearest"
|
| 23 |
+
},
|
| 24 |
+
"validation_processing": {
|
| 25 |
+
"rescale_only": true
|
| 26 |
+
}
|
| 27 |
+
},
|
| 28 |
+
"input_requirements": {
|
| 29 |
+
"image_formats": [
|
| 30 |
+
".jpg",
|
| 31 |
+
".jpeg",
|
| 32 |
+
".png"
|
| 33 |
+
],
|
| 34 |
+
"color_channels": 3,
|
| 35 |
+
"expected_range": [
|
| 36 |
+
0,
|
| 37 |
+
255
|
| 38 |
+
],
|
| 39 |
+
"batch_processing": true
|
| 40 |
+
},
|
| 41 |
+
"output_processing": {
|
| 42 |
+
"prediction_threshold": 0.5,
|
| 43 |
+
"class_mapping": {
|
| 44 |
+
"negative": 0,
|
| 45 |
+
"positive": 1
|
| 46 |
+
},
|
| 47 |
+
"confidence_scoring": true
|
| 48 |
+
},
|
| 49 |
+
"compatibility": {
|
| 50 |
+
"framework": "tensorflow",
|
| 51 |
+
"versions": [
|
| 52 |
+
"2.x"
|
| 53 |
+
],
|
| 54 |
+
"formats": [
|
| 55 |
+
".h5",
|
| 56 |
+
".tflite"
|
| 57 |
+
]
|
| 58 |
+
}
|
| 59 |
+
}
|