ZhengPeng7
/

BiRefNet

Image Segmentation

background-removal

mask-generation

Dichotomous Image Segmentation

Camouflaged Object Detection

Salient Object Detection

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions Community

BiRefNet / BiRefNet_config.py

ZhengPeng7's picture

For users to load in one key.

fc65c0f 5 months ago

224 Bytes

	from transformers import PretrainedConfig

	class BiRefNetConfig(PretrainedConfig):
	model_type = "SegformerForSemanticSegmentation"
	def __init__(
	self,
	**kwargs
	):
	super().__init__(**kwargs)