Commit 6b012da9 authored by Sikhin VC's avatar Sikhin VC

added file copy logic

parent 54d43dc4
...@@ -38,6 +38,7 @@ class ModelOptimization: ...@@ -38,6 +38,7 @@ class ModelOptimization:
logger.info(f"Current directory is : {current_directory}") logger.info(f"Current directory is : {current_directory}")
build_path = os.path.join(current_directory, "yolov5", "build") build_path = os.path.join(current_directory, "yolov5", "build")
if os.path.isdir('yolov5/build'): if os.path.isdir('yolov5/build'):
logger.info("build directory exists. Removing build directory!!")
shutil.rmtree('yolov5/build') shutil.rmtree('yolov5/build')
os.mkdir(build_path) os.mkdir(build_path)
weight_name_with_extension = os.path.basename(self.weight_path) weight_name_with_extension = os.path.basename(self.weight_path)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment