zhouzhouha / AutoMetric Public

forked from DeepTrial/AutoMetric

Notifications You must be signed in to change notification settings
Fork 0
Star 0

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

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
README.rst		README.rst
autoMetric.py		autoMetric.py
example.py		example.py
kaggle_ROC.png		kaggle_ROC.png
requirements.txt		requirements.txt
setup.py		setup.py

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.

Usage

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!