From 6b8a3049249f8b48eb8ce523d8672c1a924acadd Mon Sep 17 00:00:00 2001
From: Adolfhill <52448551+Adolfhill@users.noreply.github.com>
Date: Mon, 15 Jan 2024 22:23:49 +0800
Subject: [PATCH] Update test.py

the index of object150_info.csv start from 1, but pred start from 0. So it should be names[int(row[0]) - 1] instead of names[int(row[0])]
---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index a0a2eec2..b0e476e1 100644
--- a/test.py
+++ b/test.py
@@ -24,7 +24,7 @@
     reader = csv.reader(f)
     next(reader)
     for row in reader:
-        names[int(row[0])] = row[5].split(";")[0]
+        names[int(row[0]) - 1] = row[5].split(";")[0]
 
 
 def visualize_result(data, pred, cfg):