當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。