Believe this fixed multiprocessing subprocess issue, update release

thanhtuan2009 · Jul 7, 2021 · 0851d46 · 0851d46
1 parent 406d5e6
commit 0851d46
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@
 
 setup(
     name="backgroundremover",
-    version="0.1.2",
+    version="0.1.3",
     description="Background remover from image and video",
     long_description=long_description,
     long_description_content_type="text/markdown",

diff --git a/src/backgroundremover/__init__.py b/src/backgroundremover/__init__.py
@@ -4,6 +4,6 @@
 A library to remove background from videos and images
 """
 
-__version__ = "0.1.2"
+__version__ = "0.1.3"
 __author__ = 'Johnathan Nader'
 __credits__ = 'BackgroundRemover.app'
diff --git a/src/backgroundremover/utilities.py b/src/backgroundremover/utilities.py
@@ -1,6 +1,6 @@
 import os
 import math
-import multiprocessing
+import torch.multiprocessing as multiprocessing
 import subprocess as sp
 import time
 import ffmpeg
@@ -9,6 +9,10 @@
 import tempfile
 from .bg import DEVICE, Net, iter_frames, remove_many
 import shlex
+try:
+     set_start_method('spawn')
+except RuntimeError:
+    pass
 
 def worker(worker_nodes,
            worker_index,