Skip to content

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

added file copy logic

parent 3309dd26
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
yolo_optimization.py
View file @ 54d43dc4
...@@ -37,7 +37,7 @@ class ModelOptimization: ...@@ -37,7 +37,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'):
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