/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.2.2/rocAL_pybind/amd/rocal/plugin/pytorch.py File Reference#
pytorch.py File Reference
File containing iterators to be used with pytorch trainings. More...
Data Structures | |
class | rocAL_pybind.amd.rocal.plugin.pytorch.ROCALGenericIterator |
Iterator for processing data. More... | |
class | rocAL_pybind.amd.rocal.plugin.pytorch.ROCALClassificationIterator |
rocAL iterator for classification tasks for PyTorch. More... | |
class | rocAL_pybind.amd.rocal.plugin.pytorch.ROCALAudioIterator |
rocAL iterator for audio tasks for PyTorch The Tensors that are returned by the iterator will be owned by rocAL and would be valid until next iteration. More... | |
Functions | |
def | rocAL_pybind.amd.rocal.plugin.pytorch.draw_patches (img, idx, bboxes) |
Writes images to disk as a PNG file. More... | |
Detailed Description
File containing iterators to be used with pytorch trainings.
Function Documentation
◆ draw_patches()
def rocAL_pybind.amd.rocal.plugin.pytorch.draw_patches | ( | img, | |
idx, | |||
bboxes | |||
) |
Writes images to disk as a PNG file.
@param img The input image as a tensor. @param idx Index used for naming the output file. @param bboxes List of bounding boxes.