jhj0517 commited on
Commit
1de582e
·
1 Parent(s): a7f9f59

Update raise type

Browse files
Files changed (1) hide show
  1. modules/sam_inference.py +7 -4
modules/sam_inference.py CHANGED
@@ -90,6 +90,7 @@ class SamInference:
90
  )
91
  except Exception as e:
92
  logger.exception("Error while loading SAM2 model")
 
93
 
94
  def init_video_inference_state(self,
95
  vid_input: str,
@@ -128,8 +129,8 @@ class SamInference:
128
  try:
129
  generated_masks = self.mask_generator.generate(image)
130
  except Exception as e:
131
- logger.exception("Error while auto generating masks")
132
- raise f"Error while auto generating masks: str({e})"
133
  return generated_masks
134
 
135
  def predict_image(self,
@@ -154,6 +155,7 @@ class SamInference:
154
  )
155
  except Exception as e:
156
  logger.exception(f"Error while predicting image with prompt: {str(e)}")
 
157
  return masks, scores, logits
158
 
159
  def add_prediction_to_frame(self,
@@ -181,7 +183,7 @@ class SamInference:
181
  )
182
  except Exception as e:
183
  logger.exception(f"Error while predicting frame with prompt: {str(e)}")
184
- raise RuntimeError(f"Failed to predicting frame with prompt: {str(e)}") from e
185
 
186
  return out_frame_idx, out_obj_ids, out_mask_logits
187
 
@@ -211,6 +213,7 @@ class SamInference:
211
  }
212
  except Exception as e:
213
  logger.exception(f"Error while propagating in video: {str(e)}")
 
214
 
215
  return video_segments
216
 
@@ -266,7 +269,7 @@ class SamInference:
266
  ):
267
  if self.video_predictor is None or self.video_inference_state is None:
268
  logger.exception("Error while adding filter to preview, load video predictor first")
269
- raise f"Error while adding filter to preview"
270
 
271
  if not image_prompt_input_data["points"]:
272
  error_message = ("No prompt data provided. If this is an incorrect flag, "
 
90
  )
91
  except Exception as e:
92
  logger.exception("Error while loading SAM2 model")
93
+ raise RuntimeError(f"Failed to load model") from e
94
 
95
  def init_video_inference_state(self,
96
  vid_input: str,
 
129
  try:
130
  generated_masks = self.mask_generator.generate(image)
131
  except Exception as e:
132
+ logger.exception(f"Error while auto generating masks : {e}")
133
+ raise RuntimeError(f"Failed to generate masks") from e
134
  return generated_masks
135
 
136
  def predict_image(self,
 
155
  )
156
  except Exception as e:
157
  logger.exception(f"Error while predicting image with prompt: {str(e)}")
158
+ raise RuntimeError(f"Failed to predict image with prompt") from e
159
  return masks, scores, logits
160
 
161
  def add_prediction_to_frame(self,
 
183
  )
184
  except Exception as e:
185
  logger.exception(f"Error while predicting frame with prompt: {str(e)}")
186
+ raise RuntimeError(f"Failed to predicting frame with prompt") from e
187
 
188
  return out_frame_idx, out_obj_ids, out_mask_logits
189
 
 
213
  }
214
  except Exception as e:
215
  logger.exception(f"Error while propagating in video: {str(e)}")
216
+ raise RuntimeError(f"Failed to propagate in video") from e
217
 
218
  return video_segments
219
 
 
269
  ):
270
  if self.video_predictor is None or self.video_inference_state is None:
271
  logger.exception("Error while adding filter to preview, load video predictor first")
272
+ raise RuntimeError("Error while adding filter to preview")
273
 
274
  if not image_prompt_input_data["points"]:
275
  error_message = ("No prompt data provided. If this is an incorrect flag, "