Image Segmentation
BiRefNet
Safetensors
Transformers
background-removal
mask-generation
Dichotomous Image Segmentation
Camouflaged Object Detection
Salient Object Detection
pytorch_model_hub_mixin
model_hub_mixin
custom_code
Instructions to use ZhengPeng7/BiRefNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- BiRefNet
How to use ZhengPeng7/BiRefNet with BiRefNet:
# Option 1: use with transformers from transformers import AutoModelForImageSegmentation birefnet = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True)# Option 2: use with BiRefNet # Install from https://github.com/ZhengPeng7/BiRefNet from models.birefnet import BiRefNet model = BiRefNet.from_pretrained("ZhengPeng7/BiRefNet") - Transformers
How to use ZhengPeng7/BiRefNet with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="ZhengPeng7/BiRefNet", trust_remote_code=True)# Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("ZhengPeng7/BiRefNet", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
af31851
1
Parent(s): 79a6660
Move all BiRefNet github codes to the first level directory.
Browse files- birefnet.py +9 -9
birefnet.py
CHANGED
|
@@ -4,15 +4,15 @@ import torch.nn.functional as F
|
|
| 4 |
from kornia.filters import laplacian
|
| 5 |
from huggingface_hub import PyTorchModelHubMixin
|
| 6 |
|
| 7 |
-
from config import Config
|
| 8 |
-
from dataset import class_labels_TR_sorted
|
| 9 |
-
from models.backbones.build_backbone import build_backbone
|
| 10 |
-
from models.modules.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
|
| 11 |
-
from models.modules.lateral_blocks import BasicLatBlk
|
| 12 |
-
from models.modules.aspp import ASPP, ASPPDeformable
|
| 13 |
-
from models.modules.ing import *
|
| 14 |
-
from models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
|
| 15 |
-
from models.refinement.stem_layer import StemLayer
|
| 16 |
|
| 17 |
|
| 18 |
class BiRefNet(
|
|
|
|
| 4 |
from kornia.filters import laplacian
|
| 5 |
from huggingface_hub import PyTorchModelHubMixin
|
| 6 |
|
| 7 |
+
from .config import Config
|
| 8 |
+
from .dataset import class_labels_TR_sorted
|
| 9 |
+
from .models.backbones.build_backbone import build_backbone
|
| 10 |
+
from .models.modules.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
|
| 11 |
+
from .models.modules.lateral_blocks import BasicLatBlk
|
| 12 |
+
from .models.modules.aspp import ASPP, ASPPDeformable
|
| 13 |
+
from .models.modules.ing import *
|
| 14 |
+
from .models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
|
| 15 |
+
from .models.refinement.stem_layer import StemLayer
|
| 16 |
|
| 17 |
|
| 18 |
class BiRefNet(
|