Rohit8y commited on
Commit
80aa6a1
1 Parent(s): 83e2c83

print date

Browse files
Files changed (1) hide show
  1. app.py +3 -8
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
- # Resize image if required
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