mindspore.dataset.debug.DebugHook

class mindspore.dataset.debug.DebugHook(prev_op_name=None)

数据集管道Python调试器钩子的基类。所有用户定义的钩子行为都必须继承这个基类。

为了调试数据集管道中 map 操作的输入和输出数据，用户可以在该类的 compute 函数中添加断点，可以打印日志查看数据的类型和形状等。

参数：

• prev_op_name (str, 可选) - 上一个调试点的变换名称，默认值： None ，一般不需指定。

样例：

>>> import mindspore.dataset as ds
>>> import mindspore.dataset.debug as debug
>>>
>>> class CustomizedHook(debug.DebugHook):
...     def __init__(self):
...         super().__init__()
...
...     def compute(self, *args):
...         import pdb
...         pdb.set_trace()
...         print("Data after decode", *args)
...         return args
>>>
>>> # Enable debug mode
>>> ds.config.set_debug_mode(True, debug_hook_list=[CustomizedHook()])
>>>
>>> # Define dataset pipeline
>>> dataset = ds.ImageFolderDataset(dataset_dir="/path/to/image_folder_dataset_directory")
>>> # Insert debug hook after `Decode` operation.
>>> dataset = dataset.map([vision.Decode(), CustomizedHook(), vision.CenterCrop(100)])

compute(*args)

定义该调试钩子的行为，此方法需要被子类重写。参考上述样例如何自定义一个调试钩子。

参数：

• args (Any) - 被调试的变换的输入/输出数据，可以直接打印查看。