changing show plot

arshinmar · May 28, 2018 · 583a533 · 583a533
1 parent f751787
commit 583a533
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/Evaluator.py b/lib/Evaluator.py
@@ -96,7 +96,7 @@ def GetPascalVOCMetrics(self, dictGroundTruth, dictDetected, IOUThreshold=0.5):
             ret.append(r)
         return ret
 
-    def PlotPrecisionRecallCurve(self, classId, dictGroundTruth, dictDetected, IOUThreshold=0.5, showAP=False, showInterpolatedPrecision=False, savePath=None):
+    def PlotPrecisionRecallCurve(self, classId, dictGroundTruth, dictDetected, IOUThreshold=0.5, showAP=False, showInterpolatedPrecision=False, savePath=None, showGraphic=True):
         results = self.GetPascalVOCMetrics(dictGroundTruth, dictDetected, IOUThreshold)
 
         result = None
@@ -128,8 +128,9 @@ def PlotPrecisionRecallCurve(self, classId, dictGroundTruth, dictDetected, IOUTh
         plt.legend(shadow=True)
         plt.grid()
         if savePath != None:
-            plt.savefig(savePath)            
-        plt.show()
+            plt.savefig(savePath)
+        if showGraphic == True:
+            plt.show()
 
         ret ={}
         ret['class'] = classId