README.md

---
license: mit
language:
- en
base_model:
- google/deeplabv3_mobilenet_v2_1.0_513
pipeline_tag: image-to-image
tags:
- segmentation
- sticker
- deeplab
- image
- ai-sticker
---
# 🖼️ AI Sticker Generator API

Welcome to the **AI Sticker Generator API**! This API is designed to transform images into high-quality "stickers" by isolating the primary object using advanced **semantic segmentation** techniques. The stickers produced have smooth, feathered edges to ensure a polished and professional look.

## Examples

### Before and After Transformation

- **Before:** ![Before](output/check3.png)  
  **After:** ![After](output/sticker_check3.png)

- **Before:** ![Before](output/check.png)  
  **After:** ![After](output/sticker_check.png)

- **Before:** ![Before](output/check5.png)  
  **After:** ![After](output/sticker_check5.png)

## 📋 Key Features

### 🔍 Semantic Segmentation
Automatically identifies and isolates the main subject in an image, providing precise cutouts for clear and visually appealing stickers.

### 🌟 Feathered Edges
Applies a Gaussian blur to the mask edges, creating a soft and natural transition to transparency for a more polished finish.

### ⚡ Built with FastAPI
Utilizes FastAPI for high performance, scalability, and fast response times suitable for production environments.

### 📂 Versatile Image Support
Supports **PNG** and **JPEG** formats, ensuring compatibility with widely used image types.

## 🚀 Quick Start Guide

Entire Code is on GitHub as well : ![URL]https://github.com/rajasami156/AI-Image-Sticker-Generator.git

1. **Clone the repository** and navigate to the project directory.
2. **Install dependencies** (requires Python 3.8+).
3. **Start the API server** using the provided configuration file.
4. Once the server is running, the API is accessible locally, ready to accept image uploads for sticker generation.
   

## 🛠️ API Endpoints

### `POST /create_sticker/`
- **Description**: Upload an image to generate a sticker with a transparent background.
- **Supported File Types**: PNG and JPEG formats.
- **Response**: Returns a PNG image of the sticker with a transparent background. In case of an unsupported file format, an error message will be returned.

## 🧩 How It Works

1. **Model & Preprocessing**: Uploaded images are preprocessed and passed through a pre-trained model for semantic segmentation.
2. **Mask Generation**: A binary mask isolates the main object in the image.
3. **Edge Feathering**: A Gaussian blur is applied to mask edges, creating a smooth transition.
4. **Sticker Creation**: The mask adds transparency, producing an image that can be directly used as a sticker.

## 📂 Directory Structure

Generated stickers are saved in a designated output directory, ensuring easy access and organization of created stickers.

## ⚙️ Configuration

Before running the application, confirm that the output directory exists. This directory is essential for storing generated stickers for easy retrieval and management.

## 📜 License

Licensed under the MIT License, allowing easy adaptation and building upon this work.

## 🙋‍♂️ Contributing

Contributions are welcome! To contribute, create a new issue or pull request for bug fixes, enhancements, or new features. All contributions should adhere to the project's coding standards and guidelines.

## Built with ❤️ by [SAMIULLAH]

## 📞 Support

For support or inquiries, please contact nicesami156@gmail.com.