Commit ·
a7e29d3
1
Parent(s): 6f22264
Ensemble BiRefNet codes to HF to load model with weights in one line.
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ from models.birefnet import BiRefNet
|
|
| 35 |
birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
|
| 36 |
```
|
| 37 |
|
| 38 |
-
#### Use codes +
|
| 39 |
> Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
|
| 40 |
|
| 41 |
```python
|
|
@@ -44,7 +44,7 @@ from transformers import AutoModelForImageSegmentation
|
|
| 44 |
birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
|
| 45 |
```
|
| 46 |
|
| 47 |
-
#### Use the
|
| 48 |
```python
|
| 49 |
# Imports
|
| 50 |
from PIL import Image
|
|
|
|
| 35 |
birefnet = BiRefNet.from_pretrained('zhengpeng7/birefnet')
|
| 36 |
```
|
| 37 |
|
| 38 |
+
#### Use codes + weights from HuggingFace
|
| 39 |
> Only use the weights on HuggingFace -- Pro: No need to download BiRefNet codes manually; Con: Codes on HuggingFace might not be latest version (I'll try to keep them always latest).
|
| 40 |
|
| 41 |
```python
|
|
|
|
| 44 |
birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/birefnet', trust_remote_code=True)
|
| 45 |
```
|
| 46 |
|
| 47 |
+
#### Use the loaded BiRefNet for inference
|
| 48 |
```python
|
| 49 |
# Imports
|
| 50 |
from PIL import Image
|