TheCodez / pytorch-box2pix

WIP implementation of "Box2Pix: Single-Shot Instance Segmentation by Assigning Pixels to Object Boxes"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[WIP] pytorch-box2pix alt text

Inofficial PyTorch implementation of Box2Pix: Single-Shot Instance Segmentation by Assigning Pixels to Object Boxes (Uhrig et al., 2018).

TODO:

This is needed to get the project in a state where it can be trained:

  • mAP metric

Instance segmentation can be added later as it's just a post processing step.

Requirements

  • Install PyTorch (pytorch.org)
  • pip install -r requirements.txt (Currently requires torchvision master)
  • Download the Cityscapes dataset

Usage

Train model:

python train.py --dataset-dir 'data/cityscapes'

About

WIP implementation of "Box2Pix: Single-Shot Instance Segmentation by Assigning Pixels to Object Boxes"

License:MIT License


Languages

Language:Python 100.0%