Skip to content

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

added file copy logic

parent 6fc57d81
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
yolo_optimization.py
View file @ aee832a5
...@@ -34,7 +34,7 @@ class ModelOptimization: ...@@ -34,7 +34,7 @@ class ModelOptimization:
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"yolov5/build/{weight_name_with_extension}" src = f"/home/ilens/yolov5/build/{weight_name_with_extension}"
shutil.copy(self.weight_path, src) shutil.copy(self.weight_path, src)
current_directory = os.getcwd() current_directory = os.getcwd()
logger.info(f"Current directory is : {current_directory}") logger.info(f"Current directory is : {current_directory}")
......
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