init

Author: saleem-hadad

.gitignore

@@ -0,0 +1,4 @@
+.DS_Store
+*/.DS_Store
+/logs
+.vscode

README.md

@@ -0,0 +1 @@
+# FIREAIOT-AI

config/fyp.config

@@ -0,0 +1,187 @@
+model {
+  ssd {
+    num_classes: 1
+    box_coder {
+      faster_rcnn_box_coder {
+        y_scale: 10.0
+        x_scale: 10.0
+        height_scale: 5.0
+        width_scale: 5.0
+      }
+    }
+    matcher {
+      argmax_matcher {
+        matched_threshold: 0.5
+        unmatched_threshold: 0.5
+        ignore_thresholds: false
+        negatives_lower_than_unmatched: true
+        force_match_for_each_row: true
+      }
+    }
+    similarity_calculator {
+      iou_similarity {
+      }
+    }
+    anchor_generator {
+      ssd_anchor_generator {
+        num_layers: 6
+        min_scale: 0.2
+        max_scale: 0.95
+        aspect_ratios: 1.0
+        aspect_ratios: 2.0
+        aspect_ratios: 0.5
+        aspect_ratios: 3.0
+        aspect_ratios: 0.3333
+      }
+    }
+    image_resizer {
+      fixed_shape_resizer {
+        height: 300
+        width: 300
+      }
+    }
+    box_predictor {
+      convolutional_box_predictor {
+        min_depth: 0
+        max_depth: 0
+        num_layers_before_predictor: 0
+        use_dropout: false
+        dropout_keep_probability: 0.8
+        kernel_size: 1
+        box_code_size: 4
+        apply_sigmoid_to_scores: false
+        conv_hyperparams {
+          activation: RELU_6,
+          regularizer {
+            l2_regularizer {
+              weight: 0.00004
+            }
+          }
+          initializer {
+            truncated_normal_initializer {
+              stddev: 0.03
+              mean: 0.0
+            }
+          }
+          batch_norm {
+            train: true,
+            scale: true,
+            center: true,
+            decay: 0.9997,
+            epsilon: 0.001,
+          }
+        }
+      }
+    }
+    feature_extractor {
+      type: 'ssd_mobilenet_v2'
+      min_depth: 16
+      depth_multiplier: 1.0
+      conv_hyperparams {
+        activation: RELU_6,
+        regularizer {
+          l2_regularizer {
+            weight: 0.00004
+          }
+        }
+        initializer {
+          truncated_normal_initializer {
+            stddev: 0.03
+            mean: 0.0
+          }
+        }
+        batch_norm {
+          train: true,
+          scale: true,
+          center: true,
+          decay: 0.9997,
+          epsilon: 0.001,
+        }
+      }
+    }
+    loss {
+      classification_loss {
+        weighted_sigmoid {
+        }
+      }
+      localization_loss {
+        weighted_smooth_l1 {
+        }
+      }
+      hard_example_miner {
+        num_hard_examples: 3000
+        iou_threshold: 0.99
+        loss_type: CLASSIFICATION
+        max_negatives_per_positive: 3
+        min_negatives_per_image: 3
+      }
+      classification_weight: 1.0
+      localization_weight: 1.0
+    }
+    normalize_loss_by_num_matches: true
+    post_processing {
+      batch_non_max_suppression {
+        score_threshold: 1e-8
+        iou_threshold: 0.65
+        max_detections_per_class: 100
+        max_total_detections: 100
+      }
+      score_converter: SIGMOID
+    }
+  }
+}
+
+train_config: {
+  batch_size: 128
+  optimizer {
+    rms_prop_optimizer: {
+      learning_rate: {
+        exponential_decay_learning_rate {
+          initial_learning_rate: 0.01
+          decay_steps: 800720
+          decay_factor: 0.95
+        }
+      }
+      momentum_optimizer_value: 0.9
+      decay: 0.9
+      epsilon: 1.0
+    }
+  }
+  fine_tune_checkpoint: "brain/preTrianedModel/model.ckpt"
+  fine_tune_checkpoint_type:  "detection"
+  # Note: The below line limits the training process to 200K steps, which we
+  # empirically found to be sufficient enough to train the pets dataset. This
+  # effectively bypasses the learning rate schedule (the learning rate will
+  # never decay). Remove the below line to train indefinitely.
+  data_augmentation_options {
+    random_horizontal_flip {
+    }
+  }
+  data_augmentation_options {
+    ssd_random_crop {
+    }
+  }
+}
+
+train_input_reader: {
+  tf_record_input_reader {
+    input_path: "brain/config/train.record"
+  }
+  label_map_path: "brain/config/labels.pbtxt"
+}
+
+eval_config: {
+  num_examples: 20
+  # Note: The below line limits the evdaluation process to 10 evaluations.
+  # Remove the below line to evaluate indefinitely.
+  num_visualizations: 20
+}
+
+eval_input_reader: {
+  tf_record_input_reader {
+    input_path: "brain/config/test.record"
+  }
+  label_map_path: "brain/config/labels.pbtxt"
+  shuffle: false
+  num_readers: 1
+}

config/labels.pbtxt

@@ -0,0 +1,4 @@
+item {
+  id: 1
+  name: 'fire'
+}

config/test.csv

@@ -0,0 +1,30 @@
+filename,width,height,class,xmin,ymin,xmax,ymax
+new-188.jpg,447,299,fire,153,94,237,225
+new-86.jpg,424,500,fire,239,14,424,201
+new-92.jpg,630,442,fire,104,4,575,403
+new-177.jpg,259,194,fire,158,79,259,125
+new-79.jpg,800,535,fire,380,52,599,292
+new-79.jpg,800,535,fire,224,176,421,327
+new-79.jpg,800,535,fire,122,327,350,518
+new-163.jpg,280,180,fire,31,83,230,152
+new-45.jpg,299,168,fire,110,27,231,138
+new-51.jpg,275,183,fire,36,82,272,159
+new-201.jpg,275,183,fire,206,37,275,140
+new-201.jpg,275,183,fire,5,9,97,157
+new-201.jpg,275,183,fire,36,50,275,145
+new-215.jpg,852,480,fire,321,32,712,217
+new-215.jpg,852,480,fire,119,177,365,352
+new-229.jpg,900,600,fire,29,151,885,562
+new-7.jpg,640,360,fire,150,26,390,336
+new-6.jpg,500,375,fire,117,27,374,224
+new-228.jpg,640,360,fire,240,17,453,284
+new-214.jpg,800,535,fire,410,70,592,267
+new-214.jpg,800,535,fire,225,194,406,313
+new-214.jpg,800,535,fire,140,334,302,517
+new-214.jpg,800,535,fire,87,50,613,533
+new-200.jpg,275,183,fire,184,17,259,118
+new-50.jpg,299,168,fire,65,19,235,158
+new-44.jpg,500,377,fire,134,59,421,323
+new-162.jpg,634,391,fire,110,14,599,235
+new-78.jpg,375,500,fire,46,159,220,317
+new-78.jpg,375,500,fire,323,37,375,257