add: 背景マスキングのモデルをpickleでload

.gitattributes

@@ -0,0 +1 @@
+deeplabv3_resnet101.pkl filter=lfs diff=lfs merge=lfs -text

deeplabv3_resnet101.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d3cc589cf318a3e3b010dd2b701d38d0fbe779da680cead45e63a16a3016e08
+size 244712524

torchvision_funcs.py

@@ -1,11 +1,13 @@
 import numpy as np
 import cv2
 from PIL import Image
+import pickle
 
 import torch
 import torchvision
 from torchvision import transforms
 
+
 def deeplabv3_remove_bg(img):
     img = np.array(img, dtype=np.uint8)
     # img = cv2.imread(image_path)
@@ -14,10 +16,8 @@ def deeplabv3_remove_bg(img):
     # img = cv2.resize(img,(1000,1000))
     
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-    model = torchvision.models.segmentation.deeplabv3_resnet101(pretrained=True)
-    model = model.to(device)
-    model.eval();
+    with open('deeplabv3_resnet101.pkl', 'rb') as f:
+        model = pickle.load(f)
     
     preprocess = transforms.Compose([
     transforms.ToTensor(),