Image Segmentation
Transformers
Safetensors
cond_unet
ultrasound
medical-image-segmentation
attention-unet
custom-pipeline
custom_code
Instructions to use AImageLab-Zip/US_Cond-UNet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use AImageLab-Zip/US_Cond-UNet with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-segmentation", model="AImageLab-Zip/US_Cond-UNet", trust_remote_code=True)# Load model directly from transformers import AutoModelForImageSegmentation model = AutoModelForImageSegmentation.from_pretrained("AImageLab-Zip/US_Cond-UNet", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -37,7 +37,7 @@ from transformers import pipeline
|
|
| 37 |
|
| 38 |
segmenter = pipeline(
|
| 39 |
"image-segmentation",
|
| 40 |
-
model="
|
| 41 |
trust_remote_code=True,
|
| 42 |
)
|
| 43 |
|
|
|
|
| 37 |
|
| 38 |
segmenter = pipeline(
|
| 39 |
"image-segmentation",
|
| 40 |
+
model="AImageLab-Zip/US_Cond-UNet",
|
| 41 |
trust_remote_code=True,
|
| 42 |
)
|
| 43 |
|