SeanSdahl / PytorchDataloaderForTensorflow

This repository defines a python class that can be used to load data for the tf.keras.model.fit_generator function by using a torch.utils.data.Dataloader object for image data. This way transforms on the input image data can be transformed using the PyTorch library but still be used to fit a tf.keras.model.
11Updated 3 weeks ago

Related projects

Alternatives and complementary repositories for PytorchDataloaderForTensorflow