Upload README.md
Browse files
README.md
CHANGED
|
@@ -62,3 +62,49 @@ train_results = model.train(
|
|
| 62 |
|
| 63 |
## Test data
|
| 64 |
If you do not have test images, I placed 4 test images from the khan dataset to be used during your testing phase.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
|
| 63 |
## Test data
|
| 64 |
If you do not have test images, I placed 4 test images from the khan dataset to be used during your testing phase.
|
| 65 |
+
Test the model [here](https://huggingface.co/spaces/UEmmanuel5/profsam-fire-demo)
|
| 66 |
+
|
| 67 |
+
|
| 68 |
+
## ⚖️ License
|
| 69 |
+
|
| 70 |
+
* **Model weights** (`Fire_best.pt`): **AGPL-3.0** (Ultralytics-trained).
|
| 71 |
+
* **Space code** (this repo): **Apache-2.0**.
|
| 72 |
+
|
| 73 |
+
---
|
| 74 |
+
|
| 75 |
+
## 🙏 Acknowledgements
|
| 76 |
+
|
| 77 |
+
* Ultralytics YOLO11
|
| 78 |
+
* SAM / SAM2 ecosystem and community
|
| 79 |
+
|
| 80 |
+
---
|
| 81 |
+
|
| 82 |
+
## 📚 Citation
|
| 83 |
+
|
| 84 |
+
If this demo or the model is useful in your research, please cite:
|
| 85 |
+
|
| 86 |
+
**Manuscript**
|
| 87 |
+
|
| 88 |
+
```bibtex
|
| 89 |
+
@inproceedings{profsam2025,
|
| 90 |
+
author = {Emmanuel U. Ugwu and Xinming Zhang},
|
| 91 |
+
title = {Promptable Fire Segmentation: Unleashing SAM2's Potential for Real-Time Mobile Deployment with Strategic Bounding Box Guidance},
|
| 92 |
+
booktitle = {ICIGP '26},
|
| 93 |
+
year = {2026},
|
| 94 |
+
address = {Wuhan, China},
|
| 95 |
+
month = jan,
|
| 96 |
+
note = {to appear}
|
| 97 |
+
}
|
| 98 |
+
```
|
| 99 |
+
|
| 100 |
+
**Model/Code**
|
| 101 |
+
|
| 102 |
+
```bibtex
|
| 103 |
+
@software{profsam2025,
|
| 104 |
+
author = {Ugwu, Emmanuel U. and Zhang, Xinming},
|
| 105 |
+
title = {Promptable Fire Segmentation: Unleashing SAM2’s Potential for Real-Time Mobile Deployment with Strategic Bounding Box Guidance},
|
| 106 |
+
year = {2025},
|
| 107 |
+
doi = {10.5281/zenodo.17340313},
|
| 108 |
+
url = {https://doi.org/10.5281/zenodo.17340313}
|
| 109 |
+
}
|
| 110 |
+
```
|