当前位置: 首页>>编程示例 >>用法及示例精选 >>正文


Python mxnet.image.ImageDetIter.draw_next用法及代码示例

用法:

draw_next(color=None, thickness=2, mean=None, std=None, clip=True, waitKey=None, window_name='draw_next', id2labels=None)

参数

  • color(tuple) - RGB 中的边界框颜色,使用 None 作为随机颜色
  • thickness(int) - 边界框边框厚度
  • mean(True or numpy.ndarray) - 补偿均值以获得更好的视觉效果
  • std(True or numpy.ndarray) - 恢复标准差
  • clip(bool) - 如果为 true,则剪辑到 [0, 255] 以获得更好的视觉效果
  • waitKey(None or int) - 如果设置,则保持窗口等待 waitKey 毫秒,如果没有则跳过绘图
  • window_name(str) - 如果设置了 waitKey,则绘制窗口名称。
  • id2labels(dict) - 标签 id 到标签名称的映射。

返回

返回类型

numpy.ndarray

显示下一个带有边界框的图像。

例子

>>> # use draw_next to get images with bounding boxes drawn
>>> iterator = mx.image.ImageDetIter(1, (3, 600, 600), path_imgrec='train.rec')
>>> for image in iterator.draw_next(waitKey=None):
...     # display image
>>> # or let draw_next display using cv2 module
>>> for image in iterator.draw_next(waitKey=0, window_name='disp'):
...     pass

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.image.ImageDetIter.draw_next。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。