Commit bb4f9abc authored by Maria Kleppestø Mcculloch's avatar Maria Kleppestø Mcculloch
Browse files

Update main.py

parent 0308208b
......@@ -8,35 +8,6 @@ from qualityEvaluation import ValueHandling as QualityTemp
from utils.baseScore import BaseHandler
import matplotlib.pyplot as mpl
'''
Perform Quality Assesment
Calcualting the Baseline scores:
QualEval = QualityEvaluator("ArcFaceCasiaV2", projectpath, imgsdir, "subjectnr/samplenr")
QualEval.ArcFace_init()
QualEval.readInDataset(params, projectpath, imgsdir, "ArcFaceCasia", True )
QualEval.ComputeBaselines()
QualEval.SaveBaslines()
Computing the quality metrics:
QualEval = QualityEvaluator("ArcFaceCasiaV2", projectpath, imgsdir, "subjectnr/samplenr")
QualEval.Detection_Arcface_init()
QualEval.ComputeMetric("sharpness", params[0], params[1])
QualEval.ComputeMetric("focus", params[0], params[1])
QualEval.SaveMetric("sharpness")
QualEval.SaveMetric("focus")
Plotting:
QualEval = QualityEvaluator("ArcFaceCasiaV2", projectpath, imgsdir, "subjectnr/samplenr")
QualEval.PlotMetric("EDC", "focus",subnr=params[0], sampnr=params[1])
QualEval.PlotMetric("EDC", "sharpness",subnr=params[0], sampnr=params[1])
QualEval.PlotMetric("EDC", "brisque",subnr=params[0], sampnr=params[1], metric_name="data/brisque/quality_values_aug")
QualEval.PlotMetric("DET", "focus",subnr=params[0], sampnr=params[1])
QualEval.PlotMetric("DET", "sharpness",subnr=params[0], sampnr=params[1])
QualEval.PlotMetric("DET", "brisque",subnr=params[0], sampnr=params[1], metric_name="data/brisque/quality_values_aug")
QualEval.PlotDETCurve(params[0],params[1])
'''
# Init vars
#params = [51, 5]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment