Cleaned up maskrcnn implementation

Author: jsbroks

app/api/models.py

@@ -5,8 +5,14 @@
 from ..config import Config
 from PIL import Image
 
-if len(Config.MASK_RCNN_FILE) != 0:
+import os
+
+MODEL_LOADED = len(Config.MASK_RCNN_FILE) != 0 and os.path.isfile(Config.MASK_RCNN_FILE)
+
+if MODEL_LOADED:
     from ..util.mask_rcnn import model as maskrcnn
+else:
+    print("MaskRCNN model is disabled.", flush=True)
 
 api = Namespace('model', description='Model related operations')
 
@@ -21,8 +27,8 @@ class MaskRCNN(Resource):
     @api.expect(image_upload)
     def post(self):
         """ COCO data test """
-        if len(Config.MASK_RCNN_FILE) == 0:
-            return {"coco": {}}
+        if not MODEL_LOADED:
+            return {"disabled": True, "coco": {}}
 
         args = image_upload.parse_args()
         im = Image.open(args.get('image'))

app/util/mask_rcnn.py

@@ -1,9 +1,9 @@
-from keras.preprocessing.image import img_to_array
 from ..config import Config as AnnotatorConfig
 from skimage.transform import resize
 import imantics as im
 import cv2
 
+from keras.preprocessing.image import img_to_array
 from mrcnn.config import Config
 import mrcnn.utils as utils
 import mrcnn.model as modellib

docker-compose.dev.yml

@@ -35,25 +35,25 @@ services:
       - FILE_WATCHER=true
       - FLASK_APP=app
       - FLASK_ENV=development
-      - MASK_RCNN_FILE=/models/mask_rcnn_coco.h5
-      # Default coco classes (order matters)
-      - |
-        MASK_RCNN_CLASSES=
-        BG,person,bicycle,car,motorcycle,airplane,
-        bus,train,truck,boat,traffic light,
-        fire hydrant,stop sign,parking meter,bench,bird,
-        cat,dog,horse,sheep,cow,elephant,bear,
-        zebra,giraffe,backpack,umbrella,handbag,tie,
-        suitcase,frisbee,skis,snowboard,sports ball,
-        kite,baseball bat,baseball glove,skateboard,
-        surfboard,tennis racket,bottle,wine glass,cup,
-        fork,knife,spoon,bowl,banana',apple,
-        sandwich,orange,broccoli,carrot,hot dog,pizza,
-        donut,cake,chair,couch,potted plant,bed,
-        dining table,toilet,tv,laptop,mouse,remote,
-        keyboard,cell phone,microwave,oven,toaster,
-        sink,refrigerator,book,clock,vase,scissors,
-        teddy bear,hair drier,toothbrush
+      # - MASK_RCNN_FILE=/models/mask_rcnn_coco.h5
+      # # Default coco classes (order matters)
+      # - |
+      #   MASK_RCNN_CLASSES=
+      #   BG,person,bicycle,car,motorcycle,airplane,
+      #   bus,train,truck,boat,traffic light,
+      #   fire hydrant,stop sign,parking meter,bench,bird,
+      #   cat,dog,horse,sheep,cow,elephant,bear,
+      #   zebra,giraffe,backpack,umbrella,handbag,tie,
+      #   suitcase,frisbee,skis,snowboard,sports ball,
+      #   kite,baseball bat,baseball glove,skateboard,
+      #   surfboard,tennis racket,bottle,wine glass,cup,
+      #   fork,knife,spoon,bowl,banana',apple,
+      #   sandwich,orange,broccoli,carrot,hot dog,pizza,
+      #   donut,cake,chair,couch,potted plant,bed,
+      #   dining table,toilet,tv,laptop,mouse,remote,
+      #   keyboard,cell phone,microwave,oven,toaster,
+      #   sink,refrigerator,book,clock,vase,scissors,
+      #   teddy bear,hair drier,toothbrush
     networks:
       - cocoannotator
     depends_on: