time limit exceeds on the model, so we remove all the unnecessary saving of files, updated the code and save time

Author: one2clouds

inference_docker.py

@@ -356,7 +356,7 @@ def forward(self, **kwargs) -> torch.Tensor:
         print(np.unique(overall_mask, return_counts=True))
 
         os.makedirs(join(OUTPUT_PATH, 'images/pelvic-fracture-ct-segmentation'), exist_ok=True)
-        sitk.WriteImage(sitk.GetImageFromArray(overall_mask), join(OUTPUT_PATH, 'images/pelvic-fracture-ct-segmentation', 'overall_pred.mha'), useCompression=True)
+        sitk.WriteImage(sitk.GetImageFromArray(overall_mask), join(OUTPUT_PATH, 'images/pelvic-fracture-ct-segmentation', 'output.mha'), useCompression=True)
 
     # # Save your output
     # # write_array_as_image_file(