kitlucid commited on
Commit
8421df6
·
verified ·
1 Parent(s): 276d404

CI: deploy Docker/PDM Space

Browse files
Files changed (1) hide show
  1. services/app_service/deim_model.py +1 -3
services/app_service/deim_model.py CHANGED
@@ -109,7 +109,6 @@ class DeimHgnetV2MDrone(BaseModel):
109
  if not mask.any():
110
  return np.zeros((0, 5), dtype=np.float32)
111
  valid = b_np[mask]
112
- valid = valid
113
  cx, cy, box_w, box_h = valid[:, 0], valid[:, 1], valid[:, 2], valid[:, 3]
114
  x1 = cx - box_w / 2
115
  y1 = cy - box_h / 2
@@ -198,7 +197,6 @@ class DeimHgnetV2MDrone(BaseModel):
198
  out_dir.mkdir(parents=True, exist_ok=True)
199
  out_path = out_dir / f"{input_p.stem}_annotated.mp4"
200
 
201
-
202
  try:
203
  while True:
204
  ret, frame_bgr = cap.read()
@@ -226,5 +224,5 @@ class DeimHgnetV2MDrone(BaseModel):
226
 
227
  if __name__ == "__main__":
228
  model = DeimHgnetV2MDrone(version="v2")
229
- output_image = model.predict_video("/Users/rimma_vakhreeva/PycharmProjects/NATSAR-Demo/services/app_service/resources/videos/raw/sample1.mp4", 0.3)
230
  output_image.show()
 
109
  if not mask.any():
110
  return np.zeros((0, 5), dtype=np.float32)
111
  valid = b_np[mask]
 
112
  cx, cy, box_w, box_h = valid[:, 0], valid[:, 1], valid[:, 2], valid[:, 3]
113
  x1 = cx - box_w / 2
114
  y1 = cy - box_h / 2
 
197
  out_dir.mkdir(parents=True, exist_ok=True)
198
  out_path = out_dir / f"{input_p.stem}_annotated.mp4"
199
 
 
200
  try:
201
  while True:
202
  ret, frame_bgr = cap.read()
 
224
 
225
  if __name__ == "__main__":
226
  model = DeimHgnetV2MDrone(version="v2")
227
+ output_image = model.predict_video("./resources/videos/raw/sample1.mp4", 0.3)
228
  output_image.show()