Startseite Erkunden Hilfe
Anmelden
dyadmisha
/
AI-on-the-edge-device
Mirror von https://github.com/jomjol/AI-on-the-edge-device
1
0
Fork 0
Dateien Issues 0 Wiki
Quellcode durchsuchen

add example UI

CaCO3 vor 2 Jahren
Ursprung
716dcb5a84
Commit
a5d2086159
4 geänderte Dateien mit 55 neuen und 9 gelöschten Zeilen
Geteilte Ansicht Diff-Statistik anzeigen
  1. 17 0
      code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.cpp
  2. 3 1
      code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.h
  3. 7 5
      code/components/jomjol_flowcontroll/ClassFlowControll.cpp
  4. 28 3
      sd-card/html/websocket.html

+ 17 - 0
code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.cpp
Datei anzeigen 

@@ -37,6 +37,23 @@ ClassFlowCNNGeneral::ClassFlowCNNGeneral(ClassFlowAlignment *_flowalign, t_CNNTy
 
     imagesRetention = 5;
 
 }
 
 
+ClassFlowCNNGeneral::ClassFlowCNNGeneral(ClassFlowAlignment *_flowalign, std::string _cnn_name) : ClassFlowImage(NULL, TAG)
 
+{
 
+    string cnnmodelfile = "";
 
+    modelxsize = 1;
 
+    modelysize = 1;
 
+    CNNGoodThreshold = 0.0;
 
+    ListFlowControll = NULL;
 
+    previousElement = NULL;   
+    SaveAllFiles = false; 
+    disabled = false;
 
+    isLogImageSelect = false;
 
+    CNNType = AutoDetect;
 
+    flowpostalignment = _flowalign;
 
+    imagesRetention = 5;
 
+    cnn_name = _cnn_name;
 
+}
 
+
 
 
 string ClassFlowCNNGeneral::getReadout(int _analog = 0, bool _extendedResolution, int prev, float _before_narrow_Analog, float analogDigitalTransitionStart)
 
 {

+ 3 - 1
code/components/jomjol_flowcontroll/ClassFlowCNNGeneral.h
Datei anzeigen 

@@ -25,6 +25,7 @@ protected:
 
     t_CNNType CNNType;
 
     std::vector<general*> GENERAL;
 
     float CNNGoodThreshold;
 
+    std::string cnn_name = "ClassFlowCNNGeneral";
 
 
 	//moved to define.h
 
     //float Analog_error = 3.0;
 
@@ -56,6 +57,7 @@ protected:
 
 
 public:
 
     ClassFlowCNNGeneral(ClassFlowAlignment *_flowalign, t_CNNType _cnntype = AutoDetect);
 
+    ClassFlowCNNGeneral(ClassFlowAlignment *_flowalign, std::string _cnn_name);
 
 
     bool ReadParameter(FILE* pfile, string& aktparamgraph);
 
     bool doFlow(string time);
 
@@ -81,7 +83,7 @@ public:
 
 
     t_CNNType getCNNType(){return CNNType;};
 
 
-    string name(){return "ClassFlowCNNGeneral";}; 
+    string name(){return "ClassFlowCNNGeneral " + cnn_name;}; 
 };
 
 
 #endif

+ 7 - 5
code/components/jomjol_flowcontroll/ClassFlowControll.cpp
Datei anzeigen 

@@ -86,8 +86,10 @@ std::string ClassFlowControll::TranslateAktstatus(std::string _input)
 
         return ("Take Image");
 
     if (_input.compare("ClassFlowAlignment") == 0)
 
         return ("Aligning");
 
-    if (_input.compare("ClassFlowCNNGeneral") == 0)
 
-        return ("Digitalization of ROIs");
 
+    if (_input.compare("ClassFlowCNNGeneral Digital") == 0)
 
+        return ("Digitalization of Digital ROIs");
 
+    if (_input.compare("ClassFlowCNNGeneral Analog") == 0)
 
+        return ("Digitalization of Analog ROIs");
 
     #ifdef ENABLE_MQTT
 
         if (_input.compare("ClassFlowMQTT") == 0)
 
             return ("Sending MQTT");
 
@@ -101,7 +103,7 @@ std::string ClassFlowControll::TranslateAktstatus(std::string _input)
 
     if (_input.compare("ClassFlowPostProcessing") == 0)
 
         return ("Post-Processing");
 
 
-    return "Unkown Status";
 
+    return "Unkown Status: " + _input +"";
 
 }
 
 
 
@@ -223,12 +225,12 @@ ClassFlow* ClassFlowControll::CreateClassFlow(std::string _type)
 
     }
 
     if (toUpper(_type).compare("[ANALOG]") == 0)
 
     {
 
-        cfc = new ClassFlowCNNGeneral(flowalignment);
 
+        cfc = new ClassFlowCNNGeneral(flowalignment, std::string("Analog"));
 
         flowanalog = (ClassFlowCNNGeneral*) cfc;
 
     }
 
     if (toUpper(_type).compare(0, 7, "[DIGITS") == 0)
 
     {
 
-        cfc = new ClassFlowCNNGeneral(flowalignment);
 
+        cfc = new ClassFlowCNNGeneral(flowalignment, std::string("Digital"));
 
         flowdigit = (ClassFlowCNNGeneral*) cfc;
 
     }
 
     #ifdef ENABLE_MQTT

+ 28 - 3
sd-card/html/websocket.html
Datei anzeigen 

@@ -10,7 +10,11 @@
 
 </head>
 
 <body>
 
 
-<div id="rx"></div>
 
+<div id="state">State: Please wait...</div>
 
+<hr>
 
+<div id="img">Image: Please wait...</div>
 
+<hr>
 
+<div id="rx">Events:<br></div>
 
 
 
 <script>
 
@@ -45,7 +49,14 @@
 
   
   function onMessage(event) {
 
       console.log(event);
 
-      addToLog(event.data);              
+      addToLog(event.data);   
+      
+      data = JSON.parse(event.data)
 
+
 
+      // State
 
+      if (data.hasOwnProperty("state")) {
 
+        processStateChange(data.state);
 
+      }
 
   }
 
   
   
@@ -56,9 +67,23 @@
 
   
   
   function addToLog(msg) {
 
-      document.getElementById('rx').innerHTML += "[" + new Date().toLocaleTimeString() + "] " +msg + "<br>\n";
 
+      document.getElementById('rx').innerHTML += "[" + new Date().toLocaleTimeString() + "] " + msg + "<br>\n";
 
       window.scrollBy(0,document.body.scrollHeight);
 
   }
 
+
 
+  function processStateChange(state) {
 
+    document.getElementById('state').innerHTML = "State: <b>" + data.state + "</b>";
 
+
 
+    // TODO only reload image for those steps where it changed...
 
+    var d = new Date();
 
+    var timestamp = d.getTime();
 
+    document.getElementById('img').innerHTML = 'Image: <br><img src=' + getDomainname() + '/img_tmp/alg_roi.jpg?timestamp='+ timestamp +'" height=200px;"></img>';
 
+  }
 
+
 
+
 
+  function updateImage() {
 
+
 
+  }
 
   
 </script>
 
 </body>