run training inference with with nosave argument

zauberzeug · Nov 2, 2023 · f52b6dd · f52b6dd
1 parent 3cb13f4
commit f52b6dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trainer/app_code/yolov5_trainer.py b/trainer/app_code/yolov5_trainer.py
@@ -182,7 +182,7 @@ async def _detect(self, model_information: ModelInformation, images:  List[str],
         if self.is_cla:
             cmd = f'python /app/pred_cla.py --weights {model_folder}/model.pt --source {images_folder} --img-size {img_size} --save-txt'
         else:
-            cmd = f'python /app/pred_det.py --weights {model_folder}/model.pt --source {images_folder} --img-size {img_size} --conf-thres 0.2 --save-txt --save-conf'
+            cmd = f'python /app/pred_det.py --weights {model_folder}/model.pt --source {images_folder} --img-size {img_size} --conf-thres 0.2 --save-txt --save-conf --nosave'
         logging.info(f'running detection with command :\n {cmd}')
 
         executor.start(cmd)