nicer

Dockerfile

@@ -40,6 +40,6 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-#CMD ["python", "main.py"]
+CMD ["python", "setup.py build develop --user"]
+
 CMD ["python", "app.py"]
-#RUN main.py

app.py

@@ -12,8 +12,6 @@ import warnings
 
 warnings.filterwarnings("ignore")
 
-os.system("python setup.py build develop --user")
-
 from maskrcnn_benchmark.config import cfg
 from maskrcnn_benchmark.engine.predictor_glip import GLIPDemo
 
@@ -51,7 +49,7 @@ fiber_demo = GLIPDemo(
 
 athetics_params = {
     "skip_name": False, # whether we overlay the phrase over the box
-    "override_color": (0, 0, 0), # box color, default is white
+    "override_color": (0, 90, 190),
     "text_size": 1.0,
     "text_pixel": 3,
     "box_alpha": 1.0,

maskrcnn_benchmark/engine/predictor_glip.py

@@ -153,7 +153,7 @@ class GLIPDemo(object):
             new_width = int(new_height * aspect_ratio)
             resized_image = cv2.resize(image, (new_width, new_height))
             return resized_image
-        result = resize_image_by_height(result)
+        #result = resize_image_by_height(result)
         if self.show_mask_heatmaps:
             return self.create_mask_montage(result, top_predictions)
 
@@ -269,7 +269,7 @@ class GLIPDemo(object):
         for box, color in zip(boxes, colors):
             box = box.to(torch.int64)
             top_left, bottom_right = box[:2].tolist(), box[2:].tolist()
-            new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple((255, 255, 255)), box_pixel+3)
+            new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple((255, 255, 255)), box_pixel+5)
             new_image = cv2.rectangle(new_image, tuple(top_left), tuple(bottom_right), tuple(color) if override_color is None else tuple(override_color), box_pixel)
 
         image = cv2.addWeighted(new_image, box_alpha, image, 1 - box_alpha, 0)
@@ -338,9 +338,10 @@ class GLIPDemo(object):
             if int(y) - text_offset_original < 20:
                 y += 50
             text_box_size = cv2.getTextSize(s, cv2.FONT_HERSHEY_SIMPLEX, text_size, text_pixel)[0]
-            position = (int(x), int(y)-text_offset_original)
-            bottom_left_corner_of_text = (position[0], position[1] + text_box_size[1])
-            top_right_corner = (position[0] + text_box_size[0], position[1])
+            position = (int(x), int(y)-text_offset_original-text_box_size[1])
+            offset = 5
+            bottom_left_corner_of_text = (position[0]-offset, position[1] + text_box_size[1]+offset)
+            top_right_corner = (position[0] + text_box_size[0]+offset, position[1]-offset)
 
             image[position[1]:bottom_left_corner_of_text[1], position[0]:top_right_corner[0]] = (255, 255, 255)
             cv2.putText(