model weights uploaded
Browse files- README.md +39 -3
- rescuenet_yolo_runs/.DS_Store +0 -0
- rescuenet_yolo_runs/dataset.yaml +12 -0
- rescuenet_yolo_runs/yolov10/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov10/yolov10m_bs_16_e100_full/weights/yolov10m-e100-b16-full-best.pt +3 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov5/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov8/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/weights/last.pt +3 -0
- rescuenet_yolo_runs/yolov8_seg/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8_seg/segment/.DS_Store +0 -0
- rescuenet_yolo_runs/yolov8_seg/segment/weights/best.pt +3 -0
- rescuenet_yolo_runs/yolov8_seg/segment/weights/last.pt +3 -0
README.md
CHANGED
@@ -1,3 +1,39 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# RescueNet: A High Resolution UAV Semantic Segmentation Dataset for Natural Disaster Damage Assessment
|
2 |
+
>https://www.kaggle.com/datasets/yaroslavchyrko/rescuenet
|
3 |
+
|
4 |
+
# Test the model on HF Spaces working with best weights found:
|
5 |
+
>https://huggingface.co/spaces/dolphinium/rescuenet-damaged-building-detection
|
6 |
+
|
7 |
+
# Check experiment results @cometML platform:
|
8 |
+
>https://www.comet.com/dolphinium/rescuenet-damaged-building-detection/view/new/panels
|
9 |
+
|
10 |
+
# Check documentation pdf:
|
11 |
+
> [documentation](https://github.com/dolphinium/rescuenet-damaged-building-detection/blob/main/documentation/documentation.pdf)
|
12 |
+
|
13 |
+
# Model comparison table:
|
14 |
+
![model_comparison_table](figures/model_comparison_table.jpeg)
|
15 |
+
|
16 |
+
|
17 |
+
# Technologies and frameworks used on this project:
|
18 |
+
* Yolov5-8-10
|
19 |
+
* CometML(For monitoring and maintaining models performance)
|
20 |
+
* Huggingface Spaces(For hosting and deploying models)
|
21 |
+
* Gradio(For building a web app)
|
22 |
+
* Folium(For mapping)
|
23 |
+
* PyEXIFTool(For extracting metadata from drone imagery)
|
24 |
+
|
25 |
+
|
26 |
+
# Known Issues:
|
27 |
+
* Problem with extracting metadata from images on HF Spaces platform. Image metadata extracting is working fine on local but problematic at host. Default parameter values are used for now. See issue at HF forums at following URL:
|
28 |
+
https://discuss.huggingface.co/t/image-lost-xmp-data-on-uploads/100954
|
29 |
+
|
30 |
+
|
31 |
+
|
32 |
+
|
33 |
+
## TODOS:
|
34 |
+
* Fine-tuning the model.
|
35 |
+
* Creating requirements.txt for the project.
|
36 |
+
* Editing the readme for a better documentation.
|
37 |
+
* Making the geolocations more precise.
|
38 |
+
* Cleaning up and creating a new repository for hosting.
|
39 |
+
|
rescuenet_yolo_runs/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/dataset.yaml
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# SET PATH
|
2 |
+
path: ../data
|
3 |
+
|
4 |
+
train: images/train
|
5 |
+
val: images/val
|
6 |
+
test: images/test
|
7 |
+
|
8 |
+
nc: 2
|
9 |
+
|
10 |
+
names:
|
11 |
+
0: "undamaged building"
|
12 |
+
1: "damaged building"
|
rescuenet_yolo_runs/yolov10/.DS_Store
ADDED
Binary file (8.2 kB). View file
|
|
rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb9fde61a468537c1c587349ea0c6fbbc27146156368f82111f3021d2cd60eb6
|
3 |
+
size 5752420
|
rescuenet_yolo_runs/yolov10/base_model_bs16_non_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:317ccd867a4ad05d9098759fc5d130170a54003a95afda777c008464d56af7fb
|
3 |
+
size 5755684
|
rescuenet_yolo_runs/yolov10/yolov10m_bs_16_e100_full/weights/yolov10m-e100-b16-full-best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6f6e92c43d323b232ccf4dee1c3a55118efac45fb7733fd0a952e6ad0f3aa4c7
|
3 |
+
size 33471997
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23a3ca2bc5fb2555d94a6c9ae7c888b0c88232f3baebd5fb5edfdb9ec4f7dbf5
|
3 |
+
size 16527660
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_16_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:94cb17c3ac5c7e62d31f68e17b101df6882e6eee8238be6515eb8f3f54c8f02e
|
3 |
+
size 16529516
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:51bbe10702f5632afd438b8f7262902fab26c0a76e90765bae9539044abe4798
|
3 |
+
size 16527148
|
rescuenet_yolo_runs/yolov10/yolov10s_bs_32_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5f9eb304bb312775ea6b459737b2e3dd203c105bdf8779cc06f61db10ce44efc
|
3 |
+
size 16529516
|
rescuenet_yolo_runs/yolov5/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fef4af53173a3e7d5d9b7ec1f854d24c3d8cd4a0888193d9857c504383fedcf0
|
3 |
+
size 42181524
|
rescuenet_yolo_runs/yolov5/yolov5m_bs16_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dc24ac2a4fe5606597081e0b179b2d520ffd2aa4e794ed56b69ba16d4278b5f3
|
3 |
+
size 42181524
|
rescuenet_yolo_runs/yolov8/.DS_Store
ADDED
Binary file (8.2 kB). View file
|
|
rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e7843060d1adbd021502dbcec13316fb60398777d1262e0079d9fcf226e06beb
|
3 |
+
size 22512035
|
rescuenet_yolo_runs/yolov8/yolov8n_fixed_annotation/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dfce5e60040ddddd942626fd224f3e86a698a1a8b402043e7241668f420e4e18
|
3 |
+
size 22512419
|
rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:121f39952f83685bac1a6a32ee3438e8350cad17fad4172c330a4457425c39c8
|
3 |
+
size 22512163
|
rescuenet_yolo_runs/yolov8/yolov8s_bs32_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d0469df9980f45a4f0b9a6f0d041ea6bca561d4487caf8301ef9791e20afcfdf
|
3 |
+
size 22512419
|
rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6bcfbee48a32cb75a435ab27245d800e43995393d6aa95266ac07ca81f21049f
|
3 |
+
size 22512099
|
rescuenet_yolo_runs/yolov8/yolov8s_bs64_extended/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1a3124164787343391ae15ac96b8118eefb8afca46a518b00a865f10beffab80
|
3 |
+
size 22512419
|
rescuenet_yolo_runs/yolov8_seg/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov8_seg/segment/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
rescuenet_yolo_runs/yolov8_seg/segment/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ab1ecdf1518ea8a18ea9eabc4c24e38a1ba249f1c0f9d4120cbd1b0ff51641a1
|
3 |
+
size 6778221
|
rescuenet_yolo_runs/yolov8_seg/segment/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1e36c65d978eb28a51c6df336841d563dabae93d1f83f9510ac2f9bf7199f84f
|
3 |
+
size 6778541
|