init

modnet.py

@@ -15,7 +15,7 @@ class ModNet:
         self.session = onnxruntime.InferenceSession(model_path, None)
 
     # Get x_scale_factor & y_scale_factor to resize image
-    def get_scale_factor(im_h, im_w, ref_size):
+    def get_scale_factor(self, im_h, im_w, ref_size):
 
         if max(im_h, im_w) < ref_size or min(im_h, im_w) > ref_size:
             if im_w >= im_h:
@@ -58,7 +58,7 @@ class ModNet:
         im = (im - 127.5) / 127.5
 
         im_h, im_w, im_c = im.shape
-        x, y = get_scale_factor(im_h, im_w, ref_size)
+        x, y = self.get_scale_factor(im_h, im_w, ref_size)
 
         # resize image
         im = cv2.resize(im, None, fx=x, fy=y, interpolation=cv2.INTER_AREA)