initial commit

edge_engine/model_optimizer.py

@@ -14,7 +14,7 @@ class ModelOptimization:
         logger.info(f"Provided number of classes and image size are :  {self.num_class} and {self.image_size}")
         try:
 
-            with open('yolov5/yololayer.h', 'r') as file:
+            with open('/app/edge_engine/yolov5/yololayer.h', 'r') as file:
                 # read a list of lines into data
                 data = file.readlines()
 
@@ -23,7 +23,7 @@ class ModelOptimization:
             data[21] = f"    static constexpr int INPUT_W = {self.image_size};\n"
 
             # and write everything back
-            with open('yolov5/yololayer.h', 'w') as file:
+            with open('/app/edge_engine/yolov5/yololayer.h', 'w') as file:
                 file.writelines(data)
             logger.info("Successfully changed configurations")
         except Exception as e:
@@ -37,16 +37,16 @@ class ModelOptimization:
 
             logger.info(f"Current directory is :  {current_directory}")
             build_path = os.path.join(current_directory, "yolov5", "build")
-            if os.path.isdir('yolov5/build'):
+            if os.path.isdir('/app/edge_engine/yolov5/build'):
                 logger.info("build directory exists. Removing build directory!!")
-                shutil.rmtree('yolov5/build')
+                shutil.rmtree('/app/edge_engine/yolov5/build')
             os.mkdir(build_path)
             weight_name_with_extension = os.path.basename(self.weight_path)
             weight_name, extension = os.path.splitext(weight_name_with_extension)
-            src = f"{current_directory}/yolov5/build/"
+            src = "/app/edge_engine/yolov5/build"
             shutil.copy(self.weight_path, src)
             logger.info(f"Created build folder")
-            os.chdir('yolov5/build')
+            os.chdir('/app/edge_engine/yolov5/build')
             logger.info("Running CMake command")
             subprocess.run(['cmake', '..'])
             logger.info("Running Make command")