Skip to content

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

added file copy logic

parent 689c2089
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
yolo_optimization.py
View file @ f4a27375
...@@ -20,7 +20,7 @@ class ModelOptimization: ...@@ -20,7 +20,7 @@ class ModelOptimization:
data[19] = f" static constexpr int CLASS_NUM = {self.num_class};\n" data[19] = f" static constexpr int CLASS_NUM = {self.num_class};\n"
data[20] = f" static constexpr int INPUT_H = {self.image_size};\n" data[20] = f" static constexpr int INPUT_H = {self.image_size};\n"
data[21] = f" static constexpr int INPUT_H = {self.image_size};\n" data[21] = f" static constexpr int INPUT_W = {self.image_size};\n"
# and write everything back # and write everything back
with open('yolov5/yololayer.h', 'w') as file: with open('yolov5/yololayer.h', 'w') as file:
......
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