hlydecker commited on
Commit
f38fd61
·
1 Parent(s): 4378f9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
32
 
33
  def segment_buildings(input_image):
34
 
35
- im = cv2.imread(input_image.name)
36
  outputs = predictor(im)
37
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
38
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
 
32
 
33
  def segment_buildings(input_image):
34
 
35
+ im = cv2.imread(input_image)
36
  outputs = predictor(im)
37
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
38
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))