Remove Objects Server
Table of Contents
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.
- This endpoint accepts an image file in the
http://{localhost}:{port}/mask
- Mask endpoint is used to apply a mask to an image. The route accepts
img
andmask
as input parameters. Then, it applies a mask on an image. - You can use
testX.png
image andtestX_mask.png
mask in image folder for testing.
- Mask endpoint is used to apply a mask to an image. The route accepts
License
This project is licensed under the MIT License - see the LICENSE file for details.
Other command
docker build -t zest .