/coco_unet_segmentation

Multitask problem of Segmentation for COCO dataset

Primary LanguagePython

Multiclass Segmentation using Unet for COCO dataset

Description:

Repository for training Neural Network for Multiclass task (80 classes) for coco dataset

Fast Launch instructions:

Run init.py for downloading all needed data (annotations + images)

  • main/train.py for training
  • main/predict.py for prediction
  • examples/example_load_img_and_mask.py for visualization