| { | |
| "architecture": "BriaRMBG", | |
| "family": "rmbg", | |
| "in_ch": 3, | |
| "model": { | |
| "_name_or_path": "briaai/RMBG-1.4", | |
| "architectures": [ | |
| "BriaRMBG" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "MyConfig.RMBGConfig", | |
| "AutoModelForImageSegmentation": "briarmbg.BriaRMBG" | |
| }, | |
| "custom_pipelines": { | |
| "image-segmentation": { | |
| "impl": "MyPipe.RMBGPipe", | |
| "pt": [ | |
| "AutoModelForImageSegmentation" | |
| ], | |
| "tf": [], | |
| "type": "image" | |
| } | |
| }, | |
| "in_ch": 3, | |
| "model_type": "SegformerForSemanticSegmentation", | |
| "out_ch": 1, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.38.0.dev0" | |
| }, | |
| "out_ch": 1, | |
| "preprocessor": { | |
| "do_normalize": true, | |
| "do_pad": false, | |
| "do_rescale": true, | |
| "do_resize": true, | |
| "feature_extractor_type": "ImageFeatureExtractor", | |
| "image_mean": [ | |
| 0.5, | |
| 0.5, | |
| 0.5 | |
| ], | |
| "image_std": [ | |
| 1, | |
| 1, | |
| 1 | |
| ], | |
| "resample": 2, | |
| "rescale_factor": 0.00392156862745098, | |
| "size": { | |
| "height": 1024, | |
| "width": 1024 | |
| } | |
| }, | |
| "reference_inference_input": { | |
| "mean": [ | |
| 0.5, | |
| 0.5, | |
| 0.5 | |
| ], | |
| "resize": [ | |
| 1024, | |
| 1024 | |
| ], | |
| "resize_resample": "bilinear_antialias", | |
| "std": [ | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "value_range": [ | |
| 0.0, | |
| 1.0 | |
| ] | |
| } | |
| } |