schirrmacher
commited on
Upload ./README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -1,8 +1,11 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
tags:
|
4 |
-
-
|
|
|
5 |
pretty_name: Open Remove Background Model
|
|
|
|
|
6 |
---
|
7 |
|
8 |
# Open Remove Background Model (ormbg)
|
@@ -23,10 +26,10 @@ The model was trained with the [Human Segmentation Dataset](https://huggingface.
|
|
23 |
|
24 |
After 10.000 iterations with a single NVIDIA GeForce RTX 4090 the following achievements were made:
|
25 |
|
26 |
-
- Training
|
27 |
-
- Training
|
28 |
-
- Validation
|
29 |
-
- Maximum F1
|
30 |
- Mean Absolute Error: 0.005
|
31 |
|
32 |
Output model: `/models/ormbg.pth`.
|
@@ -40,10 +43,10 @@ git clone https://github.com/xuebinqin/DIS.git
|
|
40 |
cd DIS
|
41 |
```
|
42 |
|
43 |
-
Follow the installation instructions on https://github.com/xuebinqin/DIS?tab=readme-ov-file#1-clone-this-repo
|
44 |
Download or create some data ([like this](https://huggingface.co/datasets/schirrmacher/humans)) and place it into the DIS project folder.
|
45 |
|
46 |
-
I am
|
47 |
|
48 |
- training/im (images)
|
49 |
- training/gt (ground truth)
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
tags:
|
4 |
+
- segmentation
|
5 |
+
- background remover
|
6 |
pretty_name: Open Remove Background Model
|
7 |
+
datasets:
|
8 |
+
- schirrmacher/humans
|
9 |
---
|
10 |
|
11 |
# Open Remove Background Model (ormbg)
|
|
|
26 |
|
27 |
After 10.000 iterations with a single NVIDIA GeForce RTX 4090 the following achievements were made:
|
28 |
|
29 |
+
- Training Time: 8 hours
|
30 |
+
- Training Loss: 0.1179
|
31 |
+
- Validation Loss: 0.1284
|
32 |
+
- Maximum F1 Score: 0.9928
|
33 |
- Mean Absolute Error: 0.005
|
34 |
|
35 |
Output model: `/models/ormbg.pth`.
|
|
|
43 |
cd DIS
|
44 |
```
|
45 |
|
46 |
+
Follow the installation instructions on https://github.com/xuebinqin/DIS?tab=readme-ov-file#1-clone-this-repo.
|
47 |
Download or create some data ([like this](https://huggingface.co/datasets/schirrmacher/humans)) and place it into the DIS project folder.
|
48 |
|
49 |
+
I am using the folder structure:
|
50 |
|
51 |
- training/im (images)
|
52 |
- training/gt (ground truth)
|