diff --git a/README.md b/README.md index d2479cd..5f00d40 100644 --- a/README.md +++ b/README.md @@ -3,12 +3,34 @@ This project explores the application of cloth segmentation to inpainting, allow ## Cloth Segmentation and Inpainting Labs ## Run this work ## -Jupyter notebook with the example pipeline: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/18RenTYhuPVip9SHdMLn-vnK0K57B--um#scrollTo=D0h2Y-oOCnXJ) +Jupyter notebook with the example pipeline: + +[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/18RenTYhuPVip9SHdMLn-vnK0K57B--um#scrollTo=D0h2Y-oOCnXJ) # Key Points +![](https://github.com/banyapon/StableDiffusionInpaint-ClothSegments/blob/main/images/screen1.png?raw=true) ## Pre-trained Model: This code leverages a pre-trained model, which means you don't need to train the model from scratch. You can immediately use it for inference (making predictions). ## Image Segmentation: The primary goal is to identify and segment different clothing items within an image. ## Custom Library: The functions from iglovikov_helper_functions seem to be part of a custom or external library designed to simplify common tasks in deep learning projects. + +```bash +pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting") +``` +This line of code is the heart of setting up the Stable Diffusion model for image inpainting tasks within your Python environment. + +![](https://github.com/banyapon/StableDiffusionInpaint-ClothSegments/blob/main/images/complete.jpg?raw=true) + +## Retrieves: +It fetches the Stable Diffusion Inpainting model (weights, architecture, configuration) that has been trained by RunwayML specifically for the task of filling in missing or masked areas of images. +## Sets up: +It creates an instance of the StableDiffusionInpaintPipeline class and initializes it with the downloaded model. +## Makes it usable: +It assigns this pipeline to the pipe variable, allowing you to easily call functions on pipe to perform inpainting on your images. + + +## References +- https://github.com/mberkay0 +- https://github.com/ternaus/cloths_segmentation