Skip to content

Commit

Permalink
added to_dict attribute to PanopticaResult class
Browse files Browse the repository at this point in the history
  • Loading branch information
JosefBuchner committed Nov 6, 2023
1 parent 5a9e04f commit e44cd00
Showing 1 changed file with 16 additions and 10 deletions.
26 changes: 16 additions & 10 deletions panoptica/result.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,17 +54,23 @@ def __str__(self):
)

def to_dict(self):
return (
f"Number of instances in prediction: {self.num_pred_instances}\n"
f"Number of instances in reference: {self.num_ref_instances}\n"
f"True Positives (tp): {self.tp}\n"
f"False Positives (fp): {self.fp}\n"
f"False Negatives (fn): {self.fn}\n"
f"Recognition Quality / F1 Score (RQ): {self.rq}\n"
f"Segmentation Quality (SQ): {self.sq} ± {self.sq_sd}\n"
f"Panoptic Quality (PQ): {self.pq}\n"
f"volumetric instance-wise DICE: {self.instance_dice} ± {self.instance_dice_sd}"
return ({
"Number of instances in prediction": self.num_pred_instances,
"Number of instances in reference": self.num_ref_instances,
"True Positives (tp)": self.tp,
"False Positives (fp)": self.fp,
"False Negatives (fn)": self.fn,
"Recognition Quality / F1 Score (RQ)": self.rq,
"Segmentation Quality (SQ)": self.sq,
"Segmentation Quality SD": self.sq_sd,
"Panoptic Quality (PQ)": self.pq,
"volumetric instance-wise DICE": self.instance_dice,
"volumetric instance-wise DICE SD": self.instance_dice_sd
}
)

def to_dict2(self):
return("123")

@property
def num_ref_instances(self) -> int:
Expand Down

0 comments on commit e44cd00

Please sign in to comment.