From 56b3a58ef322297b41acd727d65ce686ff7fea5e Mon Sep 17 00:00:00 2001
From: Alex_996 <45281765+koking0@users.noreply.github.com>
Date: Thu, 2 Mar 2023 21:26:47 +0800
Subject: [PATCH] Update test.py

fix error
---
 src/test.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/test.py b/src/test.py
index 1d08044dc..dd62999d8 100755
--- a/src/test.py
+++ b/src/test.py
@@ -33,8 +33,8 @@ def __getitem__(self, index):
     img_path = os.path.join(self.img_dir, img_info['file_name'])
     image = cv2.imread(img_path)
     images, meta = {}, {}
-    for scale in opt.test_scales:
-      if opt.task == 'ddd':
+    for scale in self.opt.test_scales:
+      if self.opt.task == 'ddd':
         images[scale], meta[scale] = self.pre_process_func(
           image, scale, img_info['calib'])
       else:
@@ -123,4 +123,4 @@ def test(opt):
   if opt.not_prefetch_test:
     test(opt)
   else:
-    prefetch_test(opt)
\ No newline at end of file
+    prefetch_test(opt)