Update README

README.md

@@ -74,15 +74,15 @@ C --> F([H-mean])
     ```
 3. 计算指标
     ```python
-    from text_det_metric import DetectionIoUEvaluator
+    from text_det_metric import TextDetMetric
 
-    metric = DetectionIoUEvaluator()
+    metric = TextDetMetric()
     pred_path = "pred.txt"
     metric = metric(pred_path)
     print(metric)
 
     ```
 4. 得到结果
     ```bash
-    {'precision': 0.8300988648846577, 'recall': 0.8659281894576012, 'hmean': 0.8476350719760704, 'avg_elapse': 0.22463044243038827}
+    {'precision': 0.8301, 'recall': 0.8659, 'hmean': 0.8476, 'avg_elapse': 0.2246}
     ```

compute_metric.py

@@ -1,9 +1,9 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: [email protected]
-from text_det_metric import DetectionIoUEvaluator
+from text_det_metric import TextDetMetric
 
-metric = DetectionIoUEvaluator()
+metric = TextDetMetric()
 
 pred_path = "pred.txt"
 metric = metric(pred_path)

text_det_metric/__init__.py

@@ -1,6 +1,6 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: [email protected]
-from .main import DetectionIoUEvaluator
+from .main import TextDetMetric
 
-__all__ = ["DetectionIoUEvaluator"]
+__all__ = ["TextDetMetric"]

text_det_metric/main.py

@@ -10,7 +10,7 @@
 from shapely.geometry import Polygon
 
 
-class DetectionIoUEvaluator:
+class TextDetMetric:
     def __init__(self, iou_constraint=0.5, area_precision_constraint=0.5):
         self.iou_constraint = iou_constraint
         self.area_precision_constraint = area_precision_constraint
@@ -218,7 +218,7 @@ def main():
     parser.add_argument("-pred", "--pred_path", type=str, required=True)
     args = parser.parse_args()
 
-    evaluator = DetectionIoUEvaluator()
+    evaluator = TextDetMetric()
     metrics = evaluator(args.pred_path)
     print(metrics)