mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
@@ -25,8 +25,8 @@ class Sam(nn.Module):
|
||||
image_encoder: Union[ImageEncoderViT, TinyViT],
|
||||
prompt_encoder: PromptEncoder,
|
||||
mask_decoder: MaskDecoder,
|
||||
pixel_mean: List[float] | None = None,
|
||||
pixel_std: List[float] | None = None,
|
||||
pixel_mean: List[float] = None,
|
||||
pixel_std: List[float] = None,
|
||||
) -> None:
|
||||
"""
|
||||
SAM predicts object masks from an image and input prompts.
|
||||
|
||||
Reference in New Issue
Block a user