sreenathsree1578 commited on
Commit
3433527
·
verified ·
1 Parent(s): c1f9d90

Upload 3 files

Browse files
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
+ }