Function test_image_decoder_fused

dali/test/python/decoder/test_imgcodec.py:181–216

()

def test_image_decoder_fused():
    threads = 4
    batch_size = 10
    for test_fun in [
        create_decoder_slice_pipeline,
        create_decoder_crop_pipeline,
        create_decoder_random_crop_pipeline,
    ]:
        # before CUDA 11.4 HW decoder API doesn't support ROI so we get slightly different results
        # HW decoder + slice vs fused which in this case is executed by the hybrid backend
        if test_fun == create_decoder_random_crop_pipeline or get_nvjpeg_ver() < (11, 4, 0):
            # random_resized_crop can properly handle border as it has pixels that are cropped out,
            # while plain resize following image_decoder_random_crop cannot do that
            # and must duplicate the border pixels
            def mean_close(x, y):
                return np.mean(np.abs(x - y) < 0.5)

            validation_fun = mean_close
        else:

            def mean_close(x, y):
                return np.allclose(x, y)

            validation_fun = mean_close
        for device in ["cpu", "mixed"]:
            for img_type in test_good_path:
                yield (
                    run_decode_fused,
                    test_fun,
                    good_path,
                    img_type,
                    batch_size,
                    device,
                    threads,
                    validation_fun,
                )


def check_FastDCT_body(batch_size, img_type, device):