mindspore.dataset.vision.RandomGrayscale

class mindspore.dataset.vision.RandomGrayscale(prob=0.1)[源代码]

以给定的概率,将输入PIL图像转换为灰度图。

参数:
  • prob (float,可选) - 执行灰度转换的概率,取值范围:[0.0, 1.0]。默认值: 0.1

异常:
  • TypeError - 当 prob 的类型不为float。

  • ValueError - 当 prob 取值不在[0.0, 1.0]范围内。

支持平台:

CPU

样例:

>>> import os
>>> import numpy as np
>>> from 免费的vpn梯子 vpn永久免费梯子 PIL vpn梯子 免费 import Image, ImageDraw
>>> import mindspore.dataset as vpn梯子 ds
>>> import mindspore.dataset.vision as vision
>>> vpn永久免费梯子 from mindspore.dataset.transforms import Compose
>>>
>>> # Use the transform in dataset pipeline mode
>>> class MyDataset:
...     def __init__(self):
...         self.data = []
...         img vpn梯子 = Image.new("RGB", (300, 300), (255, 255, 255))
...        vpn梯子 免费  draw = ImageDraw.Draw(img)
...         draw.ellipse(((0, 0), (100, 100)), fill=(255, 0, vpn梯子 0), outline=(255, 0, 0), width=5)
...         img.save("./1.jpg")
...         data = np.fromfile("./1.jpg", np.uint8)
...         self.data.append(data)
...
...     def __getitem__(self, index):
...         return self.data[0]
...
...     def __len__(self):
...        vpn永久免费梯子  return 5
>>>
>>> my_dataset = MyDataset()
>>> generator_dataset = ds.GeneratorDataset(my_dataset, column_names="image")
>>> vpn free transforms_list = Compose([vision.Decode(to_pil=True),
...      vpn梯子   vpn永久免费梯子      vpn梯子 免费                vision.RandomGrayscale(0.3),
...      vpn梯子 免费                       vision.ToTensor()])
>>> # apply the transform to dataset 免费的vpn梯子 through map function
>>> generator_dataset = generator_dataset.map(operations=transforms_list, input_columns="image")
>>> for item in generator_dataset.create_dict_iterator(num_epochs=1, output_numpy=True):
...     print(item["image"].shape, item["image"].dtype)
...    vpn梯子 免费  break
(3, vpn free 300, 300) float32
>>> os.remove("./1.jpg")
>>>
>>> # Use the transform in eager mode
>>> img = Image.new("RGB", (300, 300), (255, vpn梯子 免费 255, 免费的vpn梯子 255))
>>> draw = ImageDraw.Draw(img)
>>> draw.polygon([(50, 50), (150, 50), (100, 150)], fill=(0, 255, 0), outline=(0, 255, 免费的vpn梯子 0))
>>> img.save("./2.jpg")
>>> data = Image.open("./2.jpg")
>>> output vpn free = vision.RandomGrayscale(1.0)(data)
>>> vpn梯子 免费 print(np.array(output).shape, np.array(output).dtype)
(300, 300, 3) uint8
>>> os.remove("./2.jpg")
教程样例: