Skip to content

draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem

Notifications You must be signed in to change notification settings

ChengruZhu/AutoMetric

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

autometric

autometric is to help draw ROC,PR curves for image semantic segmentation problem.When you have groundtruth image and predict probabilistic image,this package can draw ROC,PR curves and save them to your savepath.It can also calculate FPR (False Positive Rate) TPR(True Positive Rate) AUC(Area under Curve) MAP(Mean Average Precision) IoU(Intersection over Union) at the same time.

Install

Now you can use pip to install autometric,try pip install autometric

requirements:

  • python 3.0+
  • opencv_python
  • matplotlib
  • numpy

description

autometric include the following function/class:

  • CollectData: a class to calculate FPR,TPR,AUC,MAP,IOU
  • DrawCurve: a class to draw and save ROC PR curves
  • fileList: a function to collect file to a list
  • drawCurve: a function to draw curves and calculate parameters.Read example.py to get an idea of how to use!

if you right use the package,you can see the roc curve like below:

kaggle_ROC.png

About

draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%