(
data_path, device, use_fast_idct=False, memory_stats=False, jpeg_fancy_upsampling=False
)
| 49 | |
| 50 | @pipeline_def |
| 51 | def decoder_pipe( |
| 52 | data_path, device, use_fast_idct=False, memory_stats=False, jpeg_fancy_upsampling=False |
| 53 | ): |
| 54 | inputs, labels = fn.readers.file(file_root=data_path, shard_id=0, num_shards=1, name="Reader") |
| 55 | decoded = fn.decoders.image( |
| 56 | inputs, |
| 57 | device=device, |
| 58 | output_type=types.RGB, |
| 59 | use_fast_idct=use_fast_idct, |
| 60 | memory_stats=memory_stats, |
| 61 | jpeg_fancy_upsampling=jpeg_fancy_upsampling, |
| 62 | ) |
| 63 | |
| 64 | return decoded, labels |
| 65 | |
| 66 | |
| 67 | test_data_root = get_dali_extra_path() |
no test coverage detected