abc

yolo_optimization.py

@@ -36,6 +36,8 @@ class ModelOptimization:
     def optimize_model(self, weight_path):
         try:
             current_directory = os.getcwd()
+            if os.path.exists(os.path.join(current_directory, 'tensorrtx/yolov5/build')):
+
                 shutil.rmtree(os.path.join(current_directory, 'tensorrtx/yolov5/build'))
             os.mkdir(os.path.join(current_directory, 'tensorrtx/yolov5/build'))
             shutil.copy(weight_path, os.path.join(current_directory, 'tensorrtx/yolov5/build'))