Skip to content

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

api changes

parent f04b8b6b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
yolo_optimization.py
View file @ 55220b73
...@@ -46,7 +46,7 @@ class ModelOptimization: ...@@ -46,7 +46,7 @@ class ModelOptimization:
shutil.copy(self.weight_path, os.path.join(code_directory, 'tensorrtx/yolov5/build')) shutil.copy(self.weight_path, os.path.join(code_directory, 'tensorrtx/yolov5/build'))
logger.info("Copied wts file to build directory") logger.info("Copied wts file to build directory")
weight_name_with_extension = os.path.basename(self.weight_path) weight_name_with_extension = os.path.basename(self.weight_path)
os.chdir('tensorrtx/yolov5/build') os.chdir(os.path.join(code_directory, 'tensorrtx/yolov5/build'))
logger.info("Changed to build directory") logger.info("Changed to build directory")
logger.info("Running CMake command") logger.info("Running CMake command")
subprocess.run(['cmake', '..']) subprocess.run(['cmake', '..'])
......
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