schirrmacher commited on
Commit
d7b2280
1 Parent(s): 150d962

Upload ./README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +10 -7
README.md CHANGED
@@ -1,8 +1,11 @@
1
  ---
2
  license: apache-2.0
3
  tags:
4
- - art
 
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 time: 8 hours
27
- - Training loss 0.1179
28
- - Validation loss: 0.1284
29
- - Maximum F1 score: 0.9928
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 applying the folder structure:
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)