Spaces:
Sleeping
Sleeping
print date
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import os
|
2 |
|
3 |
os.system('python -m pip install \'git+https://github.com/facebookresearch/detectron2.git\'')
|
@@ -14,6 +15,7 @@ from detectron2.data import MetadataCatalog
|
|
14 |
|
15 |
|
16 |
def predict(input_image):
|
|
|
17 |
# Convert PIL Image to numpy array
|
18 |
input_image = np.array(input_image)
|
19 |
# Initialise model
|
@@ -32,14 +34,7 @@ def predict(input_image):
|
|
32 |
out = v.draw_panoptic_seg_predictions(panoptic_seg.to("cpu"), segments_info)
|
33 |
segmented_image = out.get_image()[:, :, ::-1]
|
34 |
|
35 |
-
|
36 |
-
if not segmented_image.shape[:2] == (height, width):
|
37 |
-
segmented_image = cv2.resize(segmented_image, (height, width))
|
38 |
-
|
39 |
-
# Combine the segmented and original image
|
40 |
-
combined_image = cv2.hconcat([segmented_image, input_image])
|
41 |
-
|
42 |
-
return combined_image
|
43 |
|
44 |
|
45 |
# Create Gradio interface
|
|
|
1 |
+
import datetime
|
2 |
import os
|
3 |
|
4 |
os.system('python -m pip install \'git+https://github.com/facebookresearch/detectron2.git\'')
|
|
|
15 |
|
16 |
|
17 |
def predict(input_image):
|
18 |
+
print(datetime.datetime.now())
|
19 |
# Convert PIL Image to numpy array
|
20 |
input_image = np.array(input_image)
|
21 |
# Initialise model
|
|
|
34 |
out = v.draw_panoptic_seg_predictions(panoptic_seg.to("cpu"), segments_info)
|
35 |
segmented_image = out.get_image()[:, :, ::-1]
|
36 |
|
37 |
+
return segmented_image
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
|
39 |
|
40 |
# Create Gradio interface
|