rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator Class Reference

• rocAL_pybind
• amd
• rocal
• plugin
• pytorch
• ROCALGenericIterator

Public Member Functions | Data Fields

rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator Class Reference

Iterator for processing data. More...

Inheritance diagram for rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator:

Public Member Functions
def	__init__ (self, pipeline, tensor_layout=types.NCHW, reverse_channels=False, multiplier=[1.0, 1.0, 1.0], offset=[0.0, 0.0, 0.0], tensor_dtype=types.FLOAT, device="cpu", device_id=0, display=False)
def	next (self)
def	__next__ (self)
def	reset (self)
def	__iter__ (self)
def	__len__ (self)
def	__del__ (self)

Data Fields
	loader
	tensor_format
	multiplier
	offset
	reverse_channels
	tensor_dtype
	device
	device_id
	batch_size
	labels_size
	output_list
	output_memory_type
	iterator_length
	display
	eos
	index
	num_batches
	last_batch_policy
	last_batch_size
	output_tensor_list
	labels_tensor
	bbox_list
	labels_list
	labels
	bboxes
	img_size
	label_2d_numpy
	bb_2d_numpy
	bb_padded
	labels_padded

Detailed Description

Iterator for processing data.

    @param pipeline            The rocAL pipeline to use for processing data.
    @param tensor_layout       The layout of the output tensors
    @param reverse_channels    Whether to reverse the order of color channels.
    @param multiplier          Multiplier values for color normalization.
    @param offset              Offset values for color normalization.
    @param tensor_dtype        Data type of the output tensors.
    @param display             Whether to display images during processing
    @param device              The device to use for processing
    @param device_id           The ID of the device to use

---

The documentation for this class was generated from the following file:

• /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.4.1/rocAL_pybind/amd/rocal/plugin/pytorch.py