fix(weighting): add simple normalization to chunk search

gnes-ai · Jul 22, 2019 · d3fbbca · d3fbbca
1 parent 87cb91c
commit d3fbbca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gnes/preprocessor/image/segmentation.py b/gnes/preprocessor/image/segmentation.py
@@ -83,5 +83,6 @@ def _crop_image_reshape(self, original_image, coordinates):
         return np.array(original_image.crop(coordinates).resize((self.target_img_size,
                                                                  self.target_img_size)))
 
-    def _cal_area(self, coordinate):
+    @classmethod
+    def _cal_area(cls, coordinate):
         return (coordinate[2] - coordinate[0]) * (coordinate[3] - coordinate[1])