/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.