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

• rocAL_pybind
• amd
• rocal
• plugin

Data Structures | Functions | Variables

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.