up

inference.py

@@ -192,7 +192,7 @@ class ModelFactory:
         
         cfg = ModelConfig(ModelCategory.ONE_STEP_OBJECT_DETECTION, 
                                          model_path="COCO-Detection/retinanet_R_101_FPN_3x.yaml",
-                                         cfg_path="../configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml").get_cfg()
+                                         cfg_path="./configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml").get_cfg()
         p = self.get_instance(ModelCategory.ONE_STEP_OBJECT_DETECTION,cfg)
 
         if input_image is None and image_path is not None:
@@ -209,7 +209,7 @@ class ModelFactory:
         """
         cfg = ModelConfig(ModelCategory.OBJECT_DETECTION, 
                                          model_path="COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml",
-                                         cfg_path="../configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml").get_cfg()
+                                         cfg_path="./configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml").get_cfg()
         p = self.get_instance(ModelCategory.IMAGE_FEATURE_EXTRACT, cfg)
         if input_image is None and image_path is not None:
             input_image = p.read_image(image_path)
@@ -225,7 +225,7 @@ class ModelFactory:
         """
         cfg = ModelConfig(ModelCategory.INSTANCE_SEGMENTATION, 
                                          model_path="COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml",
-                                         cfg_path="../configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml").get_cfg()
+                                         cfg_path="./configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml").get_cfg()
         p = self.get_instance(ModelCategory.INSTANCE_SEGMENTATION,cfg)
         if input_image is None and image_path is not None:
             input_image = p.read_image(image_path)
@@ -242,7 +242,7 @@ class ModelFactory:
 
         cfg = ModelConfig(ModelCategory.SEMANTIC_SEGMENTATION, 
                                          model_path=None,
-                                         cfg_path="../configs/PascalVOC-Detection/faster_rcnn_R_50_FPN.yaml").get_cfg()
+                                         cfg_path="./configs/PascalVOC-Detection/faster_rcnn_R_50_FPN.yaml").get_cfg()
     
         p = self.get_instance(ModelCategory.SEMANTIC_SEGMENTATION,cfg)
 
@@ -260,7 +260,7 @@ class ModelFactory:
 
         cfg = ModelConfig(ModelCategory.INSTANCE_SEGMENTATION, 
                                          model_path="COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml",
-                                         cfg_path="../configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml").get_cfg()
+                                         cfg_path="./configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml").get_cfg()
         p = self.get_instance(ModelCategory.INSTANCE_SEGMENTATION,cfg)
         if input_image is None and image_path is not None:
             input_image = p.read_image(image_path)
@@ -278,7 +278,7 @@ class ModelFactory:
 
         cfg = ModelConfig(ModelCategory.KEYPOINTS, 
                                          model_path="COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x.yaml",
-                                         cfg_path="../configs/COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x.yaml").get_cfg()
+                                         cfg_path="./configs/COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x.yaml").get_cfg()
         p = self.get_instance(ModelCategory.KEYPOINTS,cfg)
         if input_image is None and image_path is not None:
             input_image = p.read_image(image_path)

requirements.txt

@@ -10,4 +10,5 @@ fvcore==0.1.5.post20221221
 omegaconf==2.3.0
 pycocotools==2.0.7
 gradio_image_prompter==0.1.0
+cloudpickle==2.2.1
 git+https://github.com/facebookresearch/segment-anything.git