Skip to content

Commit

Permalink
spellcheck visualize.py
Browse files Browse the repository at this point in the history
  • Loading branch information
coreyhu authored and waleedka committed Jul 12, 2018
1 parent 05f2755 commit 1ac88be
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions mrcnn/visualize.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@

import os
import sys
import logging
import random
import itertools
import colorsys
Expand Down Expand Up @@ -41,7 +40,7 @@ def display_images(images, titles=None, cols=4, cmap=None, norm=None,
cols: number of images per row
cmap: Optional. Color map to use. For example, "Blues".
norm: Optional. A Normalize instance to map values to colors.
interpolation: Optional. Image interporlation to use for display.
interpolation: Optional. Image interpolation to use for display.
"""
titles = titles if titles is not None else [""] * len(images)
rows = len(images) // cols + 1
Expand Down Expand Up @@ -325,7 +324,7 @@ def plot_overlaps(gt_class_ids, pred_class_ids, pred_scores,
gt_class_ids: [N] int. Ground truth class IDs
pred_class_id: [N] int. Predicted class IDs
pred_scores: [N] float. The probability scores of predicted classes
overlaps: [pred_boxes, gt_boxes] IoU overlaps of predictins and GT boxes.
overlaps: [pred_boxes, gt_boxes] IoU overlaps of predictions and GT boxes.
class_names: list of all class names in the dataset
threshold: Float. The prediction probability required to predict a class
"""
Expand Down Expand Up @@ -361,7 +360,7 @@ def plot_overlaps(gt_class_ids, pred_class_ids, pred_scores,
def draw_boxes(image, boxes=None, refined_boxes=None,
masks=None, captions=None, visibilities=None,
title="", ax=None):
"""Draw bounding boxes and segmentation masks with differnt
"""Draw bounding boxes and segmentation masks with different
customizations.
boxes: [N, (y1, x1, y2, x2, class_id)] in image coordinates.
Expand All @@ -370,7 +369,7 @@ def draw_boxes(image, boxes=None, refined_boxes=None,
masks: [N, height, width]
captions: List of N titles to display on each box
visibilities: (optional) List of values of 0, 1, or 2. Determine how
prominant each bounding box should be.
prominent each bounding box should be.
title: An optional title to show over the image
ax: (optional) Matplotlib axis to draw on.
"""
Expand Down

0 comments on commit 1ac88be

Please sign in to comment.