Upload ./README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -29,11 +29,7 @@ python utils/inference.py
|
|
| 29 |
|
| 30 |
## Training
|
| 31 |
|
| 32 |
-
The model was trained with the [Human Segmentation Dataset](https://huggingface.co/datasets/schirrmacher/humans) which was created with [LayerDiffuse](https://github.com/layerdiffusion/LayerDiffuse) and [IC-Light](https://github.com/lllyasviel/IC-Light).
|
| 33 |
-
|
| 34 |
-
The model was trained with 10.000 iterations and on a NVIDIA GeForce RTX 4090.
|
| 35 |
-
|
| 36 |
-
Output model: `/models/ormbg.pth`.
|
| 37 |
|
| 38 |
## Want to train your own model?
|
| 39 |
|
|
|
|
| 29 |
|
| 30 |
## Training
|
| 31 |
|
| 32 |
+
The model was trained on a NVIDIA GeForce RTX 4090 (10.000 iterations) with the [Human Segmentation Dataset](https://huggingface.co/datasets/schirrmacher/humans) which was created with [LayerDiffuse](https://github.com/layerdiffusion/LayerDiffuse) and [IC-Light](https://github.com/lllyasviel/IC-Light).
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
|
| 34 |
## Want to train your own model?
|
| 35 |
|