Remove unused imports

Author: jsbroks

backend/config/config.py

@@ -61,3 +61,6 @@ class Config:
     MASK_RCNN_CLASSES = os.getenv("MASK_RCNN_CLASSES", "BG")
 
     DEXTR_FILE = os.getenv("DEXTR_FILE", "/models/dextr_pascal-sbd.h5")
+
+
+__all__ = ["Config"]

backend/database/tasks.py

@@ -93,4 +93,7 @@ def api_json(self):
         return {
             "id": self.id,
             "name": self.name
-        }
+        }
+
+
+__all__ = ["TaskModel"]

backend/webserver/api/annotations.py

@@ -1,6 +1,5 @@
 from flask_restplus import Namespace, Resource, reqparse
 from flask_login import login_required, current_user
-from imantics import Color
 
 from database import AnnotationModel
 from ..util import query_util
@@ -67,9 +66,7 @@ def get(self, annotation_id):
 
         if annotation is None:
             return {"message": "Invalid annotation id"}, 400
-
-        image = current_user.images.filter(id=annotation.image_id, deleted=False).first()
-
+        
         return query_util.fix_ids(annotation)
 
     @login_required

backend/webserver/api/info.py

@@ -1,8 +1,7 @@
-from flask_restplus import Namespace, Resource, reqparse
+from flask_restplus import Namespace, Resource
 
 from workers.tasks import long_task
 from config import Config
-from ..util.version_util import get_tag
 from database import UserModel, TaskModel
 
 

backend/webserver/api/models.py

@@ -49,8 +49,8 @@ def post(self, image_id):
 
         args = dextr_args.parse_args()
         points = args.get('points')
-        padding = args.get('padding')
-        threshold = args.get('threshold')
+        # padding = args.get('padding')
+        # threshold = args.get('threshold')
 
         if len(points) != 4:
             return {"message": "Invalid points entered"}, 400

backend/webserver/api/tasks.py

@@ -1,8 +1,7 @@
-from flask_restplus import Namespace, Resource, reqparse
-from flask_login import login_required, current_user
+from flask_restplus import Namespace, Resource
+from flask_login import login_required
 
 from ..util import query_util
-from config import Config
 from database import TaskModel
 
 

backend/webserver/authentication.py

@@ -1,12 +1,11 @@
-from flask_login import LoginManager, login_user, login_required, logout_user, current_user, AnonymousUserMixin
+from flask_login import LoginManager, AnonymousUserMixin
 from database import (
     UserModel,
     DatasetModel,
     CategoryModel,
     AnnotationModel,
     ImageModel
 )
-from config import Config
 
 login_manager = LoginManager()
 

backend/webserver/sockets.py

@@ -1,5 +1,4 @@
 import functools
-import json
 import time
 
 from flask import session

backend/webserver/util/dextr.py

@@ -1,6 +1,5 @@
 from dextr import DEXTR
 from ..config import Config
-import os
 
 
 model = DEXTR(nb_classes=1, resnet_layers=101, input_shape=(512, 512), weights_path=Config.DEXTR_FILE,

backend/webserver/util/mask_rcnn.py

@@ -1,13 +1,10 @@
 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
-import numpy as np
 
 
 MODEL_DIR = "/workspace/models"
@@ -44,7 +41,7 @@ def __init__(self):
 
     def detect(self, image):
 
-        if self.model == None:
+        if self.model is None:
             return {}
 
         image = image.convert('RGB')
@@ -56,7 +53,6 @@ def detect(self, image):
 
         masks = result.get('masks')
         class_ids = result.get('class_ids')
-        scores = result.get('scores')
 
         coco_image = im.Image(width=width, height=height)