Method test_to_keras

tests/test_loaders.py:92–99

(self)

        loader_test = loader_train.copy(ds_test, drop_transforms=True)
    
    def test_to_keras(self):
        loader = nt.Loader(self.dataset_2d, images_per_batch=4)
        keras_loader = loader.to_keras()
        x_batch, y_batch = next(iter(keras_loader))
        self.assertEqual(x_batch.shape, (4, 256, 256, 1))
        
        gen = record_generator(loader)
        xb,yb = next(iter(gen))
        
    def test_keras_multi(self):
        img2d = ants.image_read(ants.get_data('r16'))