Skip to content

segmind/segmind-py

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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") image
Stable Diffusion v2.1 SD2_1(api_key).generate("calico cat wearing a cosmonaut suit, 3d render, pixar style, 8k, high resolution") image
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") image
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") image
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") image
ControlNet Scribble ControlNet(api_key).generate(prompt = "steampunk underwater helmet, dark ocean background", imageUrl = "https://segmind.com/scribble-input.jpeg", option="Scribble") image
ControlNet Soft Edge ControlNet(api_key).generate(prompt = "royal chamber with fancy bed", imageUrl = "https://segmind.com/soft-edge-input.jpeg", option="SoftEdge") image
ControlNet Depth ControlNet(api_key).generate(prompt = "young african american man, black suit, smiling, white background", imageUrl = "https://segmind.com/depth.jpeg", option="Depth") image
ControlNet Canny ControlNet(api_key).generate(prompt = "a colorful bird, 4k", imageUrl = "https://segmind.com/canny-input.jpeg", option="Canny") image
Background Removal BackgroundRemoval(api_key).generate(imageUrl = "https://segmind.com/bg-removal.jpg") image
Face Swapper FaceSwap(api_key).generate(imageUrl = "https://segmind.com/elon.jpg", maskUrl = "https://segmind.com/burn.gif") image
Codeformer Codeformer(api_key).generate(imageUrl = "https://segmind.com/codeformer_input.png") image
ESRGAN ESRGAN(api_key).generate(imageUrl = "https://segmind.com/butterfly.png") image
SAM SAM(api_key).generate(imageUrl = "https://segmind.com/kitchen.jpg") image

Dependencies

  • PIL (Python Imaging Library)

Releases

No releases published

Packages

No packages published

Languages