Spaces:
Sleeping
Sleeping
CI: deploy Docker/PDM Space
Browse files
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("
|
| 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()
|