diff --git a/module/statistics/item.py b/module/statistics/item.py index 45c241c7c..a1594b3c6 100644 --- a/module/statistics/item.py +++ b/module/statistics/item.py @@ -40,7 +40,7 @@ class Item: self.image = image else: self.image = cv2.resize(image, self.IMAGE_SHAPE, interpolation=cv2.INTER_CUBIC) - self.is_valid = np.mean(rgb2gray(self.image) > 127) > 0.1 + self.is_valid = self.predict_valid() self._name = 'DefaultItem' self.amount = 1 self.cost = 'DefaultCost' @@ -85,6 +85,9 @@ class Item: return name + def predict_valid(self): + return np.mean(rgb2gray(self.image) > 127) > 0.1 + @property def button(self): return self._button.button