Merge pull request #48 from IndTechSpprt/result-verification

Result verification

Author: shashsuresh

.gitignore

@@ -34,3 +34,6 @@ pc_code/Algorithms/json_files/resnet18.json
 pc_code/Algorithms/json_files/test_resnet18_residual_convolution.json
 pc_code/Algorithms/json_files/test_resnet18_residual_merged.json
 pc_code/Fused/fused_layers_resnet18.json
+pc_code/Simulation/Simu_q/worker_4.json
+pc_code/Simulation/Simu_q/input_len_4.txt
+pc_code/Simulation/Simu_q/res_len_4.txt

MCU_code/PlatformIO_code/worker_code/include/communication.h

@@ -16,6 +16,7 @@
 
 /// @brief All supported message headers
 enum MessageHeaders{
+  Inference_Results     = (byte)195,
   Adaptive_Pooling      = (byte)196,
   ACK                   = (byte)197,
   Complete              = (byte)198,

MCU_code/PlatformIO_code/worker_code/src/worker_code.ino

@@ -217,11 +217,19 @@ void setup() {
           }
           count += LINEAR_SEGMENT;
         }
+        char results[MESSAGE_SIZE];
+        int res_count = 0;
         for(int k = 0; k < result_length[j]; k++){
           Serial.print(k);
           Serial.print(" ");
           Serial.println(result[k]);
+          results[RESERVED_BYTES + res_count] = result[k];
+          res_count+=1;
         }
+        results[0] = MCU_ID;
+        results[1] = Inference_Results;
+        write_length(results, res_count);
+        client.write(results, MESSAGE_SIZE);
       }
     }
   #ifdef PROFILING

MCU_code/tcp.py

@@ -62,14 +62,15 @@
 sockets = [None] * num_mcu
 addresses = [None] * num_mcu
 which = -1
+results_dict = dict()
 for count in range(num_mcu):
     # Accept a connection
     client_socket, client_address = server_socket.accept()
     client_socket.setblocking(0)  # Set socket to non-blocking mode
     print(str(client_address[0]))
     for mcu in testbed:
         if str(client_address[0]) == ip_first_part+mcu["ip_end"]:
-            which = int(mcu["mcu_id"])
+            which = int(mcu["id"])
     sockets[which] = client_socket
     addresses[which] = client_address
     print(f"Connected to Arduino {which} at:", client_address)
@@ -208,6 +209,11 @@ def send_ack(sock, data):
                         for i in range(6, 6 + len_int):
                             to_adaptive_pooling.append(received_data[i])
                         print(len(to_adaptive_pooling))
+                    
+                    elif received_data[1] == 195:
+                        len_int = struct.unpack('<I', received_data[2:6])[0]
+                        print(f"received results data from MCU{received_data[0]}, len {len_int}")
+                        results_dict[str(received_data[0])] = struct.unpack('<' + 'B'*len(received_data[6:6+len_int]), received_data[6:6+len_int])
 
                     else:
                         data_to_send = received_data
@@ -224,4 +230,11 @@ def send_ack(sock, data):
                         for w in to_which:
                             wait_for_ack(sockets[w], message_size)
 except KeyboardInterrupt:
+    index = 0
+    with open("inference_results_mcu.txt", 'w') as file:
+        for count in range(num_mcu):
+            for result in results_dict[str(count)]:
+                file.write(str(index)+": "+str(result)+"\n")
+                index+=1
+    
     print("Closing connection")

MCU_code/testbed.json

@@ -1 +1 @@
-[{"mcu_id": "0", "ip_end": "124", "mac_end": "235"}, {"mcu_id": "1", "ip_end": "123", "mac_end": "236"}, {"mcu_id": "2", "ip_end": "122", "mac_end": "237"},{"mcu_id": "3", "ip_end": "121", "mac_end": "238"}]
+[{"mcu_id": "0", "ip_end": "124", "mac_end": "235"}, {"mcu_id": "1", "ip_end": "123", "mac_end": "236"}, {"mcu_id": "2", "ip_end": "122", "mac_end": "237"}]