aerial-segmentation-model-selection

Runtime error

hlydecker commited on Sep 11, 2023

Commit

2c0f2db

•

1 Parent(s): 921eee0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@
 building-segmentation
 Proof of concept showing effectiveness of a fine tuned instance segmentation model for deteting buildings.
 """
-import os
 os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
 from transformers import DetrFeatureExtractor, DetrForSegmentation
 from PIL import Image
@@ -20,7 +20,6 @@ from detectron2.engine import DefaultPredictor
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog, DatasetCatalog
-from PIL import Image
 cfg = get_cfg()
 cfg.MODEL.DEVICE='cpu'

 building-segmentation
 Proof of concept showing effectiveness of a fine tuned instance segmentation model for deteting buildings.
 """
+import os, json, cv2, random
 os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
 from transformers import DetrFeatureExtractor, DetrForSegmentation
 from PIL import Image
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog, DatasetCatalog
 cfg = get_cfg()
 cfg.MODEL.DEVICE='cpu'