/seg_metrics_pytorch

Semantic Segmentation Metrics Functions on pytorch.[Currently for PASCAL VOC 2012, easily expandable]

Primary LanguagePython

Semantic Segmentation Metrics on Pytorch

Metrics used:

  • Pixel Accuracy
  • mean Accuracy(of per-class pixel accuracy)
  • mean IOU(of per-class Mean IOU)
  • Frequency weighted IOU

For more information, kindly refer Fully Convolutional Networks for Semantic Segmentation

Functions

Convert .mat files to .png files

Use

python demo.py convert_prediction predict_loc id_file

Convert Pascal VOC Validation filesfrom 3d-color to 2d-class-id .png format

Use

python demo.py convert_gt gt_loc id_file

Calculate the metrics

Use

python demo.py find_metrics predict_path gt_path id_file [--options]

Files

  • utils.py: contains functions.
  • demo.py: contains a brief demo of how to use the functions.[use demo.py -h]

Acknowledgement

A few parts have been adopted from the code present in martinkersner/py_img_seg_eval. Although the formulations are slightly wrong, it was very helpful.

Also, a big thanks to Video Analytics Lab.