krunakuamar/zest

Remove Objects Server

Table of Contents

1. About
2. Installation
3. Usage
4. License

About

This is a Python project for removing unwanted objects from images using the inpainting technique. It includes a server implemented with FastAPI and an endpoint for processing images by applying inpainting techniques. This project uses a deep learning library, PyTorch, for training and testing the inpainting model.

Installation

To install this project, you should first create a virtual environment using the following commands:

python3 -m venv venv
source venv/bin/activate

After creating the virtual environment, you can install the required libraries using pip:

pip install -r requirements.txt

Usage

To use this project, first start the server by running main.py:

python main.py

After the server has started, you can test following endpoints:

• http://{localhost}:{port}/lama/paint

  • This endpoint accepts an image file in the file parameter and applies inpainting techniques to remove unwanted objects.

• http://{localhost}:{port}/mask

  • Mask endpoint is used to apply a mask to an image. The route accepts img and mask as input parameters. Then, it applies a mask on an image.
  • You can use testX.png image and testX_mask.png mask in image folder for testing.

License

This project is licensed under the MIT License - see the LICENSE file for details.

Other command

docker build -t zest .