Function check_fancy_upsampling_body

dali/test/python/decoder/test_image.py:254–271

(batch_size, img_type, device)

def check_fancy_upsampling_body(batch_size, img_type, device):
    data_path = os.path.join(test_data_root, good_path, img_type)
    compare_pipelines(
        decoder_pipe(
            data_path=data_path,
            batch_size=batch_size,
            num_threads=3,
            device_id=0,
            device=device,
            jpeg_fancy_upsampling=True,
        ),
        decoder_pipe(
            data_path=data_path, batch_size=batch_size, num_threads=3, device_id=0, device="cpu"
        ),
        batch_size=batch_size,
        N_iterations=3,
        eps=1,
    )


@params(1, 8)