rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference

rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference#

rocAL: rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference
rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection Class Reference

Iterator for processing data. More...

Inheritance diagram for rocAL_pybind.amd.rocal.plugin.tf.ROCALGenericIteratorDetection:
rocAL_pybind.amd.rocal.plugin.tf.ROCALIterator

Public Member Functions

def __init__ (self, pipeline, tensor_layout=types.NCHW, reverse_channels=False, multiplier=None, offset=None, tensor_dtype=types.FLOAT, device=None, device_id=0)
 
def next (self)
 
def __next__ (self)
 
def reset (self)
 
def __iter__ (self)
 
def __del__ (self)
 

Data Fields

 loader
 
 tensor_format
 
 multiplier
 
 offset
 
 device
 
 device_id
 
 reverse_channels
 
 tensor_dtype
 
 bs
 
 output_list
 
 dimensions
 
 dtype
 
 output_tensor_list
 
 output
 
 bbox_list
 
 label_list
 
 labels
 
 bboxes
 
 img_size
 
 num_bboxes_list
 
 label_2d_numpy
 
 bb_2d_numpy
 
 target
 
 target1
 
 res
 
 l
 
 num_bboxes_arr
 

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 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.1.1/rocAL_pybind/amd/rocal/plugin/tf.py