add the script of iiit5k

Author: Charles-258

tools/eval/README.md

@@ -20,6 +20,7 @@ Supported datasets:
 - [WIDERFace](#widerface)
 - [LFW](#lfw)
 - [ICDAR](#icdar)
+- [IIIT5K](#iiit5k)
 
 ## ImageNet
 
@@ -138,6 +139,7 @@ Run evaluation with the following command:
 
 ```shell
 python eval.py -m sface -d lfw -dr /path/to/lfw
+```
 
 ## ICDAR2003
 

tools/eval/datasets/__init__.py

@@ -2,7 +2,7 @@
 from .widerface import WIDERFace
 from .lfw import LFW
 from .icdar import ICDAR
-
+from .iiit5k import IIIT5K
 
 class Registery:
     def __init__(self, name):
@@ -20,3 +20,4 @@ def register(self, item):
 DATASETS.register(WIDERFace)
 DATASETS.register(LFW)
 DATASETS.register(ICDAR)
+DATASETS.register(IIIT5K)

tools/eval/datasets/iiit5k.py

@@ -0,0 +1,55 @@
+import lmdb
+import os
+import numpy as np
+import cv2 as cv
+from tqdm import tqdm
+
+class IIIT5K:
+    def __init__(self, root):
+        self.root = root
+        self.acc = -1
+        self.inputSize = [100, 32]
+
+        self.val_label = self.load_label(self.root)
+
+    @property
+    def name(self):
+        return self.__class__.__name__
+
+    def load_label(self, root):
+        lmdb_file = root
+        lmdb_env = lmdb.open(lmdb_file)
+        lmdb_txn = lmdb_env.begin()
+        lmdb_cursor = lmdb_txn.cursor()
+        label = list()
+        for key, value in lmdb_cursor:
+            image_index = key.decode()
+            if image_index.split('-')[0] == 'image':
+                img = cv.imdecode(np.fromstring(value, np.uint8), 3)
+                label_index = 'label-' + image_index.split('-')[1]
+                value = lmdb_txn.get(label_index.encode()).decode().lower()
+                label.append([img, value])
+            else:
+                break
+        return label
+
+    def eval(self, model):
+        right_num = 0
+        pbar = tqdm(self.val_label)
+        for img, value in pbar:
+            pbar.set_description("Evaluating {} with {} val set".format(model.name, self.name))
+
+
+            rbbox = np.array([0, img.shape[0], 0, 0, img.shape[1], 0, img.shape[1], img.shape[0]])
+            pred = model.infer(img, rbbox).lower()
+            if value == pred:
+                right_num += 1
+
+        self.acc = right_num/(len(self.val_label) * 1.0)
+
+
+    def get_result(self):
+        return self.acc
+
+    def print_result(self):
+        print("Accuracy: {:.2f}%".format(self.acc*100))

tools/eval/eval.py

@@ -95,7 +95,7 @@
             target_size=112),
         icdar=dict(
             name="ICDAR",
-            topic="text_recognition")
+            topic="text_recognition"),
 )
 
 def main(args):