Skip to content

Y
yolo_model_optimization
Commit ec5f3d4f authored by Sikhin VC's avatar Sikhin VC
Browse files
Options

added file copy logic

parent c81c722f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
yolo_optimization.py
View file @ ec5f3d4f
...@@ -36,7 +36,7 @@ class ModelOptimization: ...@@ -36,7 +36,7 @@ class ModelOptimization:
current_directory = os.getcwd() current_directory = os.getcwd()
weight_name_with_extension = os.path.basename(self.weight_path) weight_name_with_extension = os.path.basename(self.weight_path)
weight_name, extension = os.path.splitext(weight_name_with_extension) weight_name, extension = os.path.splitext(weight_name_with_extension)
src = f"{current_directory}/yolov5/build/{weight_name_with_extension}" src = f"{current_directory}/yolov5/build/"
shutil.copy(self.weight_path, src) shutil.copy(self.weight_path, src)
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")
......
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