Update README.md

ashnair1 · Nov 27, 2019 · 9ffeb61 · 9ffeb61
1 parent 2f169d3
commit 9ffeb61
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -137,11 +137,10 @@ tiny
 ### 5. Generate segmentation masks
 The `cas.get_segmasks()` function allows you to create segmentation masks from your MS COCO object detection datasets. Similar to the Pascal VOC dataset, the mask values are their classes and a colour palette is applied to enable visualisation. The generated masks are stroed in the `./results` folder. Samples are shown below.
 
-|  | Detection |Segmentation   |   
-|-----|-----|---|
+|                |                            Detection                                             |                                               Segmentation                                       |   
+|----------------|----------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------|
 | **SpaceNet**   | <img src="./rep_stuff/SpaceNet.png" alt="SpaceNet" alt="SpaceNet" width=310 />   |  <img src="./rep_stuff/SpaceNet_mask.png" alt="SpaceNet_mask" title="SpaceNet_mask" width=310 /> |
-| **iSAID** | <img src="./rep_stuff/iSAID.png" alt="iSAID" title="iSAID" width=310 /> | <img src="./rep_stuff/iSAID_mask.png" alt="iSAID_mask" title="iSAID_mask" width=310 />  |
-
+| **iSAID**      | <img src="./rep_stuff/iSAID.png" alt="iSAID" title="iSAID" width=310 />          | <img src="./rep_stuff/iSAID_mask.png" alt="iSAID_mask" title="iSAID_mask" width=310 />           |
 
 ### Todo
 1.  Converter for converting COCO annotations to YOLO format.

diff --git a/coco_assistant/utils/det2seg.py b/coco_assistant/utils/det2seg.py
@@ -16,7 +16,7 @@ def det2seg(cann, output_dir):
     :param cann: COCO annotation object
     :param output_dir: Directory to store segmentation masks.
     """
-    
+
     if os.path.isdir(output_dir) is False:
         os.makedirs(output_dir, exist_ok=True)