Segmind API Wrapper
Wrapper for Segmind API for using Generative models. Visit Website
Installation
Simply Install the pip package by typing the following in the terminal:
pip install segmind
Usage
- Import Required Model Class
from segmind import Kadinsky - Instantite Model Class with your API Key
model = Kadinsky(api_key) - Generate Image
img = model.generate(prompt) - View Image
img.show()
Models Supported
Check Available Models
ControlNet
Image to Image using Stable Diffusion 1.5.
Available Options:
- Canny
- Depth
- OpenPose
- Scribble
- SoftEdge
from segmind import ControlNet
model = ControlNet(api_key)
img = model.generate(prompt, imageUrl, option)
For additional options, check Docstring of the model.
SD2_1
Text-to-image Stable diffusion 2.1 model that can generate images given a natural language prompt.
from segmind import SD2_1
model = SD2_1(api_key)
img = model.generate(prompt)
For additional options, check Docstring of the model.
Kadinsky
Image-to-image Kadinsky model that can generate images given a natural language prompt.
from segmind import Kadinsky
model = Kadinsky(api_key)
img = model.generate(prompt)
For additional options, check Docstring of the model.
SD1_5 Img2Img
A text-to-image diffusion model that can create photorealistic images from any given text input, and additionally has the ability to fill in missing parts of an image by using a mask.
from segmind import SD1_5
For additional options, check Docstring of the model.
ERSGAN
An image-to-image model that upscales low-resolution images into high-resolution ones using a GAN trained on a dataset of high-resolution images.
from segmind import ERSGAN
For additional options, check Docstring of the model.
BackgroundRemoval
The background removal model efficiently separates the main subject or the object from its surrounding background, resulting in a clean and isolated foreground.
from segmind import BackgroundRemoval
For additional options, check Docstring of the model.
Codeformer
CodeFormer is a robust face restoration algorithm for old photos or AI-generated faces.
from segmind import Codeformer
For additional options, check Docstring of the model.
SAM
Segment Anything Model (SAM) is a state-of-the-art image segmentation model that can segment any object in an image.
from segmind import SAM
model = SAM(api_key)
img = model.generate(imageUrl)
For additional options, check Docstring of the model.
FaceSwap
FaceSwap is a state-of-the-art face swapping model that can swap faces in images and videos.
from segmind import FaceSwap
model = FaceSwap(api_key)
img = model.generate(imageUrl, maskUrl)
For additional options, check Docstring of the model.
SDOutpainting
The SDOutpainting model is used for outpainting tasks, where the model is given a part of an image and it needs to generate the rest of the image.
from segmind import SDOutpainting
model = SDOutpainting(api_key)
img = model.generate(imageUrl)
For additional options, check Docstring of the model.
Word2Img
A text-to-image model that can generate images from any given text input.
from segmind import Word2Img
model = Word2Img(api_key)
img = model.generate(image, prompt)
For additional options, check Docstring of the model.
QRGenerator
A QR code generator that can generate QR codes from any given text input.
from segmind import QRGenerator
model = QRGenerator(api_key)
img = model.generate(prompt, qr_text)
For additional options, check Docstring of the model.
Text To Image
We support several text to image models:
- Stable Diffusion XL 1.0
- Segmind Tiny-SD
- Segmind Tiny-SD (Portrait)
- Segmind Small-SD
- Paragon
- Realistic Vision
- Reliberate
- Revanimated
- Colorful
- Cartoon
- Edge of Realism
- Epic Realism
- RPG
- Scifi
- Cyber Realistic
- Samaritan
- RCNZ - Cartoon
- Manmarumix
- Majicmix
- Juggernaut Final
- Icbinp
- Fruit Fusion
- Flat 2d
- Fantassified Icons
- DvArch
- Dream Shaper
- Deep Spaced Diffusion
- Cute Rich Style
- All in one pixel
- 526mix
You can check the complete list of models here.
Examples
| Model | Code Example | Generated Image |
|---|---|---|
| Stable Diffusion XL | SDXL(api_key).generate(prompt = "cinematic film still, 4k, realistic, ((cinematic photo:1.3)) of panda wearing a blue spacesuit, sitting in a bar, Fujifilm XT3, long shot, ((low light:1.4)), ((looking straight at the camera:1.3)), upper body shot, somber, shallow depth of field, vignette, highly detailed, high budget Hollywood movie, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy") |
 |
| Stable Diffusion Outpainting | SDOutpainting(api_key).generate(image = "https://www.segmind.com//image5.png", prompt = "streets in italy") |
 |
| QR Generator | QRGenerator(api_key).generate(prompt = "A mouth-watering pizza topped with gooey cheese and fresh ingredients, Close-up, Realistic Style, Art Inspirations from Food Photography", qr_text = "www.segmind.com") |
 |
| Word2Img | Word2Img(api_key).generate(image = "https://www.segmind.com//word2img_input.png", prompt = "top-view, A mouth-watering pizza topped with gooey cheese and fresh ingredients,Food Photography") |
 |
| Kadinsky | Kadinsky(api_key).generate("tiny isometric city on a tiny floating island, highly detailed, 3d render") |
|
| Stable Diffusion v2.1 | SD2_1(api_key).generate("calico cat wearing a cosmonaut suit, 3d render, pixar style, 8k, high resolution") |
|
| Stable Diffusion img2img | SD1_5(api_key).generate(prompt = "A fantasy landscape, trending on artstation, mystical sky", imageUrl= "https://segmind.com/sd-img2img-input.jpeg") |
|
| Stable Diffusion Inpainting | SD1_5(api_key).generate(prompt = "mecha robot sitting on a bench", imageUrl= "https://segmind.com/inpainting-input-image.jpeg", maskUrl= "https://segmind.com/inpainting-input-mask.jpeg") |
|
| ControlNet Openpose | ControlNet(api_key).generate(prompt = "a beautiful fashion model, wearing a red polka dress, red door background. hyperrealistic. photorealism, 4k, extremely detailed", imageUrl = "https://segmind.com/fashion2.jpeg", option="OpenPose") |
|
| ControlNet Scribble | ControlNet(api_key).generate(prompt = "steampunk underwater helmet, dark ocean background", imageUrl = "https://segmind.com/scribble-input.jpeg", option="Scribble") |
|
| ControlNet Soft Edge | ControlNet(api_key).generate(prompt = "royal chamber with fancy bed", imageUrl = "https://segmind.com/soft-edge-input.jpeg", option="SoftEdge") |
|
| ControlNet Depth | ControlNet(api_key).generate(prompt = "young african american man, black suit, smiling, white background", imageUrl = "https://segmind.com/depth.jpeg", option="Depth") |
|
| ControlNet Canny | ControlNet(api_key).generate(prompt = "a colorful bird, 4k", imageUrl = "https://segmind.com/canny-input.jpeg", option="Canny") |
|
| Background Removal | BackgroundRemoval(api_key).generate(imageUrl = "https://segmind.com/bg-removal.jpg") |
|
| Face Swapper | FaceSwap(api_key).generate(imageUrl = "https://segmind.com/elon.jpg", maskUrl = "https://segmind.com/burn.gif") |
|
| Codeformer | Codeformer(api_key).generate(imageUrl = "https://segmind.com/codeformer_input.png") |
|
| ESRGAN | ESRGAN(api_key).generate(imageUrl = "https://segmind.com/butterfly.png") |
|
| SAM | SAM(api_key).generate(imageUrl = "https://segmind.com/kitchen.jpg") |
Dependencies
- PIL (Python Imaging Library)