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):
|