/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.2/rocAL_pybind/amd/rocal/plugin/generic.py File Reference

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.2/rocAL_pybind/amd/rocal/plugin/generic.py File Reference#

rocAL: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.2/rocAL_pybind/amd/rocal/plugin/generic.py File Reference
generic.py File Reference

File containing iterators for generic use case. More...

Data Structures

class  rocAL_pybind.amd.rocal.plugin.generic.ROCALGenericIterator
 Iterator for processing data. More...
 
class  rocAL_pybind.amd.rocal.plugin.generic.ROCALClassificationIterator
 ROCAL iterator for classification tasks for generic use case. More...
 

Functions

def rocAL_pybind.amd.rocal.plugin.generic.draw_patches (img, idx)
 Writes images to disk as a PNG file. More...
 

Variables

bool rocAL_pybind.amd.rocal.plugin.generic.CUPY_FOUND = True
 

Detailed Description

File containing iterators for generic use case.

Function Documentation

◆ draw_patches()

def rocAL_pybind.amd.rocal.plugin.generic.draw_patches (   img,
  idx 
)

Writes images to disk as a PNG file.

    @param img    The input image as an array.
    @param idx    Index used for naming the output file.