Skip to content

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

added file copy logic

parent f4a27375
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
yolo_optimization.py
View file @ 3309dd26
...@@ -37,6 +37,8 @@ class ModelOptimization: ...@@ -37,6 +37,8 @@ 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'):
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)
weight_name, extension = os.path.splitext(weight_name_with_extension) weight_name, extension = os.path.splitext(weight_name_with_extension)
......
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