README.md

metadata

title: Pytorch Image Segmentation
emoji: 📊
colorFrom: pink
colorTo: pink
sdk: gradio
sdk_version: 3.22.1
app_file: app.py
pinned: false

Pytorch Image Segmentation

This repo contains the code for training a U-Net model for image segmentation on the Human Segmentation Dataset.

Usage :nut_and_bolt:

1. Clone this repo

git clone https://github.com/josebenitezg/Pytorch-Image-Segmentation

2. Create a virtual enviroment

python -m venv env

3. Activate virtual enviroment

• for linux

source env/bin/activate

• for windows

env\Scripts\Activate.bat

4. Install requirements

pip install -r requirements.txt

5. Train the model

python train.py

6. Run gradio inference app

python gradio_inference.py

This repo contains dataset files to train a small model.

Dataset Credit : https://github.com/VikramShenoy97/Human-Segmentation-Datasets