test_silhouettes.py 文件源码

def save_results(self, verbose):
        if self.tested_frame_coutner > 0:
            ave_fp = self.cum_fp / self.tested_frame_coutner
            ave_fn = self.cum_fn / self.tested_frame_coutner
            ave_wfp = self.cum_wfp / self.tested_frame_coutner
            ave_wfn = self.cum_wfn / self.tested_frame_coutner
        else:
            ave_fp = 0.
            ave_fn = 0.
            ave_wfp = 0.
            ave_wfn = 0.
        if verbose:
            print("Tested frame count: {:d}".format(self.tested_frame_coutner))
            print("Avg. false positives: {:.2f}\nAvg. false negatives: {:.2f}".format(
                ave_fp, ave_fn))
            print("Avg. weighted false positives: {:.2f}\nAvg. weighted false negatives: {:.2f}".format(
                ave_wfp, ave_wfn))
        out = {"average_false_positives": float(ave_fp),
               "average_false_negatives": float(ave_fn),
               "average_weighted_false_positives": float(ave_wfp),
               "average_weighted_false_negatives": float(ave_wfn),
               "tested_frame_count": self.tested_frame_coutner,
               "args": self.args_dict}
        out_file = open(os.path.join(self.datapath, self.output_file), "w", encoding="utf_8")
        dump(out, out_file, Dumper=Dumper)
        out_file.close()
        return 0