/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.0/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...

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.