detectron2を遅延インストール

app.py

@@ -1,3 +1,10 @@
+# https://huggingface.co/spaces/karolmajek/Detectron2-MaskRCNN/commit/0a02fceaca5ec92613aa2e8ce3b7d5e5043cbee5
+try:
+    import detectron2
+except:
+    import os 
+    os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
+
 import glob
 import gradio as gr
 from os.path import basename
@@ -6,19 +13,18 @@ from predictor import Predictor
 
 predictor = Predictor()
 
-def query_image(model: str, img_path: str, score_min: int):
+def predict(model: str, img_path: str, score_min: int):
     if not model:
         raise ValueError('モデルが指定されていません')
     if not img_path:
         return ValueError('画像が指定されていません')
-    img, count = predictor.predict(model, img_path, score_min)
-    return img, count
+    return predictor.predict(model, img_path, score_min)
 
 models = [basename(path) for path in glob.glob("models/*")]
 images = glob.glob("test/*.jpg")
 
 demo = gr.Interface(
-    query_image, 
+    predict, 
     inputs=[
         gr.Dropdown(models, label="モデル", value=models[0]),
         gr.Image(images[0], type='filepath', label="画像", interactive=False),

predictor.py

@@ -5,7 +5,7 @@ from detectron2.engine import DefaultPredictor
 from detectron2.structures import Instances
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
-from detectron2.data.datasets import register_coco_instances, load_coco_json
+from detectron2.data.datasets import load_coco_json
 
 DEVICE = 'cpu'
 
@@ -20,19 +20,9 @@ class Predictor():
         cfg = get_cfg()
         
         # 設定ファイルを取得
-        # cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
         cfg.merge_from_file(model_zoo.get_config_file("COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml"))
         cfg.MODEL.DEVICE = DEVICE
-        
-        # 学習済みのモデルファイルを指定
-        # cfg.MODEL.WEIGHTS = "models/pipe_seg_1.pth"
-        
-        # 指定した確信度以上のパイプが検知される
-        # cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.3
-        
-        # register_coco_instances("my_dataset_test", {}, "./test/_annotations.coco.json", "./test")
-        
-        cfg.DATASETS.TEST = ("my_dataset_test", )
+
         load_coco_json('./test/_annotations.coco.json', './test', 'my_dataset_test')
         test_metadata = MetadataCatalog.get("my_dataset_test")
         print(test_metadata)

requirements.txt

@@ -2,4 +2,4 @@ opencv-python-headless
 pyyaml==5.1
 torch
 torchvision
-detectron2 @ git+https://github.com/facebookresearch/detectron2.git@main
+# detectron2 @ git+https://github.com/facebookresearch/detectron2.git@main