Skip to content

Commit

Permalink
docs: added simulation results
Browse files Browse the repository at this point in the history
  • Loading branch information
kausthub-kannan committed Jul 18, 2024
1 parent 9c2f14f commit b6f1b27
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 4 deletions.
13 changes: 10 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# DROPEX
Disaster Rescue Operations and Probing using EXpert Drones is Disaster Risk Management (DRM) framework which is utilised to use swarm of drones with autonomous person detection and payload dropping making the rescue operations optimal.
<img width="600" src="assets/simulation-results.png" alt="dropex-arch"/>

## Architecture
<img width="600" src="assets/dropex-architecture.png" alt="dropex-arch"/>
Expand All @@ -11,9 +12,10 @@ These models are deployed using a FastAPI server with streamlit user interface f
[DROPEX-simulation](https://github.com/kausthub-kannan/DROPEX-simulation).

## Running simulation locally:
**Note:** The setup.py has be run at the parent directory of the project. This would also install the required DETR models
from hugging face. If you prefer to use your own models, you can replace the models in the `simulation/models` directory.
The DETR model can be found at Hugging Face: [DETR](https://huggingface.co/kausthubkannan17/dropex).
If you prefer to use your own models, you can replace the models in the `simulation/models` directory either by providing
local path or hosting your model in Hugging Face and providing the model name. The DETR model used by default can be
found at Hugging Face: [DETR](https://huggingface.co/kausthubkannan17/dropex).

```bash
cd simulation
python setup.py
Expand All @@ -30,4 +32,9 @@ cd client
streamlit run main.py
```

**Note:**
Before running the servers, add your firebase-adminsdk.json file to both `server` and `client` directories.
This is required for the firebase authentication. Also, enable read and write operations to be true in the firebase
rules for realtime database.


Binary file added assets/simulation-results.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
2 changes: 1 addition & 1 deletion simulation/client/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def show_image_from_url(url, predictions):
int(box["x2"]),
int(box["y2"]),
)
cv2.rectangle(image_cv, (x1, y1+50), (x2, y2+50), (0, 255, 0), 2)
cv2.rectangle(image_cv, (x1, y1 + 50), (x2, y2 + 50), (0, 255, 0), 2)
cv2.putText(
image_cv,
f"Class: {pred['class']} Score: {pred['score'][:5]}",
Expand Down

0 comments on commit b6f1b27

Please sign in to comment.