fix topk bug (#2945)

2023-09-01 19:28:17 +08:00 · 2023-09-01 19:28:17 +08:00 · d7a7d3e5e1
parent 74a33b7f50
commit d7a7d3e5e1
1 changed files with 3 additions and 3 deletions
--- a/ppcls/data/postprocess/topk.py
+++ b/ppcls/data/postprocess/topk.py
@ -42,9 +42,9 @@ class Topk(object):
            except Exception as e:
                with open(class_id_map_file, "r", encoding='gbk') as fin:
                    lines = fin.readlines()
-                for line in lines:
-                    partition = line.split("\n")[0].partition(self.delimiter)
-                    class_id_map[int(partition[0])] = str(partition[-1])
+            for line in lines:
+                partition = line.split("\n")[0].partition(self.delimiter)
+                class_id_map[int(partition[0])] = str(partition[-1])
        except Exception as ex:
            print(ex)
            class_id_map = None