Update Model

2023-03-01 17:16:01 +08:00
parent a40ca1f2a1
commit 7230840cf7
5 changed files with 12 additions and 4 deletions
@@ -72,6 +72,9 @@ class Plate(object):
    def to_result(self):
        return [self.plate_code, self.rec_confidence, self.plate_type, self.det_bound_box.tolist(),]

+    def to_full_result(self):
+        return [self.plate_code, self.rec_confidence, self.plate_type, self.det_bound_box.tolist(), self.vertex.tolist()]
+
    def __dict__(self):
        return self.to_dict()