提交 1ac88be8 编写于 作者: C Corey Hu 提交者: Waleed

spellcheck visualize.py

上级 05f27551
......@@ -9,7 +9,6 @@ Written by Waleed Abdulla
import os
import sys
import logging
import random
import itertools
import colorsys
......@@ -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
......@@ -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
"""
......@@ -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.
......@@ -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.
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册