From 13b8147d54f2e1890af4a49ab95346678d8551b0 Mon Sep 17 00:00:00 2001 From: LmeSzinc Date: Mon, 29 Jun 2020 16:43:24 +0800 Subject: [PATCH] Opt: Lazy template load --- module/base/template.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/module/base/template.py b/module/base/template.py index 9afee7278..17e23b916 100644 --- a/module/base/template.py +++ b/module/base/template.py @@ -12,7 +12,18 @@ class Template: """ self.server = server.server self.file = file[self.server] if isinstance(file, dict) else file - self.image = np.array(Image.open(self.file)) + self._image = None + + @property + def image(self): + if self._image is None: + self._image = np.array(Image.open(self.file)) + + return self._image + + @image.setter + def image(self, value): + self._image = value def match(self, image, similarity=0.85): """