Function test_fancy_upsampling

dali/test/python/decoder/test_imgcodec.py:273–293

(batch_size)

@params(1, 8)
def test_fancy_upsampling(batch_size):
    if get_nvjpeg_ver() < (12, 1, 0):
        raise SkipTest("nvimgcodec/nvjpeg doesn't support fancy upsampling in this version")

    data_path = os.path.join(test_data_root, good_path, "jpeg")
    compare_pipelines(
        decoder_pipe(
            data_path=data_path,
            batch_size=batch_size,
            num_threads=3,
            device_id=0,
            device="mixed",
            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,
    )


batch_size_test = 16