p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add image for imshow_det_bbox

This commit is contained in:
YangLuo 2021-12-25 16:52:32 +08:00
parent d9efe41557
commit b9096fa309
2 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
docs/api_img/dataset/browse_dataset.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 860 KiB

22
mindspore/python/mindspore/dataset/utils/browse_dataset.py
View File

@ -69,9 +69,9 @@ def imshow_det_bbox(image, bboxes, labels, segm=None, class_names=None, score_th
>>> import mindspore.dataset as ds
>>> from mindspore.dataset.utils.browse_dataset import imshow_det_bbox
>>>
>>> # Read VOC data
>>> # Read Detection dataset, such as VOC2012.
>>> voc_dataset_dir = "/path/to/voc_dataset_directory"
>>> dataset = ds.VOCDataset(voc_dataset_dir, task="Detection", shuffle=False, decode=True, num_samples=3)
>>> dataset = ds.VOCDataset(voc_dataset_dir, task="Detection", shuffle=False, decode=True, num_samples=5)
>>> dataset_iter = dataset.create_dict_iterator(output_numpy=True, num_epochs=1)
>>>
>>> # draw dataset
@ -79,22 +79,20 @@ def imshow_det_bbox(image, bboxes, labels, segm=None, class_names=None, score_th
... image = data["image"]
... bbox = data["bbox"]
... label = data["label"]
... # read your image masks if needed
... masks = np.zeros((4, image.shape[0], image.shape[1]))
... masks[0][0:500, 0:500] = 1
... masks[1][1000:1500, 1000:1500] = 2
... masks[2][0:500, 0:500] = 3
... masks[3][1000:1500, 1000:1500] = 4
... segm = masks
... # draw image with bboxes
... imshow_det_bbox(image, bbox, label, segm,
... imshow_det_bbox(image, bbox, label,
... class_names=['aeroplane', 'bicycle', 'bird', 'boat', 'bottle', 'bus', 'car', 'cat',
... 'chair', 'cow', 'diningtable', 'dog', 'horse', 'motorbike', 'person',
... 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor'],
... win_name="my_window",
... wait_time=5,
... show=False,
... wait_time=5000,
... show=True,
... out_file="voc_dataset_{}.jpg".format(str(index)))
Examples using `imshow_det_bbox` on VOC2012:
.. image:: api_img/browse_dataset.png
"""
try: