Spaces:
Sleeping
Sleeping
yisol
commited on
Commit
β’
1a13129
1
Parent(s):
595105e
update demo code
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ import apply_net
|
|
23 |
from preprocess.humanparsing.run_parsing import Parsing
|
24 |
from preprocess.openpose.run_openpose import OpenPose
|
25 |
from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
|
26 |
-
from torchvision.
|
27 |
|
28 |
|
29 |
def pil_to_binary_mask(pil_image, threshold=0):
|
|
|
23 |
from preprocess.humanparsing.run_parsing import Parsing
|
24 |
from preprocess.openpose.run_openpose import OpenPose
|
25 |
from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
|
26 |
+
from torchvision.transforms.functional import to_pil_image
|
27 |
|
28 |
|
29 |
def pil_to_binary_mask(pil_image, threshold=0):
|