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

• rocAL_pybind
• amd
• rocal
• plugin

Data Structures | Functions

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.