Skip to main content

Mask R-CNN for Fine-Grained segmentation

Project description

# fine-grained-segmentation library

Python library for segmenting clothing items in images, implemented in Python 3 and ONNX. A deep learning model generates bounding boxes and segmentation masks for each instance of an object in the image. It’s based on [Matterport Mask R-CNN](https://github.com/matterport/Mask_RCNN)

The library is also available on the [Python Package Index](https://pypi.org/project/fine-grained-segmentation/)

A demo web app is up at https://fine-grained-segmentation.vinnypalumbo.com

## Requirements

Python 3.5, ONNX runtime, and other common packages listed in requirements.txt.

## Installation

  1. Clone this repository

  2. Run setup to install the library `bash python3 setup.py install ` If it failed to install the dependencies, run `bash pip3 install -r requirements.txt `

  3. Download pre-trained weights (mrcnn.onnx) from the [releases page](https://github.com/vinny-palumbo/fine_grained_segmentation/releases)

## Usage

Here is how to use the library from the command line: `bash fashion-segmentator --image=<path/to/image/file> ` This will generate a `result.png` file in the current directory

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

fine_grained_segmentation-0.1.9.tar.gz (10.1 kB view hashes)

Uploaded Source

Built Distribution

fine_grained_segmentation-0.1.9-py3-none-any.whl (15.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page