cenkcorapci / fashion-parser

Fashion item segmentation with deep learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

fashion-parser

ForTheBadge built-with-science

License: MIT GitHub last commit

Fashion segmentation based on Matterport's mask r-cnn implementation using imaterialist-fashion-2019-FGVC6 data set.

Usage

  • Edit commons/config.py for workspace details, folder paths etc.
  • Hyper parameters are set in commons/fashion_config.py which is inherited from Matterport's config
  • Use experiment.py to train and create a submission file;
python experiment.py --epochs <number of epochs> --val_split <between 0 - 1, ratio of validation samples>

Example Results

Alt text Alt text

TODO

  • Switch to Keras
  • Change to Mask R-CNN
  • Update Readme
  • Serve with Flask

About

Fashion item segmentation with deep learning

License:MIT License


Languages

Language:Python 100.0%