Spaces:
Sleeping
Sleeping
Update binary_segmentation.py
Browse files- binary_segmentation.py +1 -0
binary_segmentation.py
CHANGED
|
@@ -17,6 +17,7 @@ import torch
|
|
| 17 |
from PIL import Image
|
| 18 |
from torchvision import transforms
|
| 19 |
import cv2
|
|
|
|
| 20 |
|
| 21 |
# Configure logging
|
| 22 |
logging.basicConfig(
|
|
|
|
| 17 |
from PIL import Image
|
| 18 |
from torchvision import transforms
|
| 19 |
import cv2
|
| 20 |
+
from u2net import U2NETP
|
| 21 |
|
| 22 |
# Configure logging
|
| 23 |
logging.basicConfig(
|