/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-rocal/checkouts/docs-6.1.1/rocAL/include/api/rocal_api_data_loaders.h Source File

• rocAL
• include
• api

rocal_api_data_loaders.h

Go to the documentation of this file.

1 /*

2 Copyright (c) 2019 - 2023 Advanced Micro Devices, Inc. All rights reserved.

3 

4 Permission is hereby granted, free of charge, to any person obtaining a copy

5 of this software and associated documentation files (the "Software"), to deal

6 in the Software without restriction, including without limitation the rights

7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

8 copies of the Software, and to permit persons to whom the Software is

9 furnished to do so, subject to the following conditions:

10 

11 The above copyright notice and this permission notice shall be included in

12 all copies or substantial portions of the Software.

13 

14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

20 THE SOFTWARE.

21 */

22 

23 #ifndef MIVISIONX_ROCAL_API_DATA_LOADERS_H

24 #define MIVISIONX_ROCAL_API_DATA_LOADERS_H

25 #include "rocal_api_types.h"

26 

50 extern "C" RocalTensor ROCAL_API_CALL rocalJpegFileSource(RocalContext context,

51  const char* source_path,

52  RocalImageColor rocal_color_format,

53  unsigned internal_shard_count,

54  bool is_output,

55  bool shuffle = false,

56  bool loop = false,

57  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

58  unsigned max_width = 0, unsigned max_height = 0, RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

59 

76 extern "C" RocalTensor ROCAL_API_CALL rocalJpegFileSourceSingleShard(RocalContext context,

77  const char* source_path,

78  RocalImageColor rocal_color_format,

79  unsigned shard_id,

80  unsigned shard_count,

81  bool is_output,

82  bool shuffle = false,

83  bool loop = false,

84  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

85  unsigned max_width = 0, unsigned max_height = 0, RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

86 

101 extern "C" RocalTensor ROCAL_API_CALL rocalSequenceReader(RocalContext context,

102  const char* source_path,

103  RocalImageColor rocal_color_format,

104  unsigned internal_shard_count,

105  unsigned sequence_length,

106  bool is_output,

107  bool shuffle = false,

108  bool loop = false,

109  unsigned step = 0,

110  unsigned stride = 0);

111 

127 extern "C" RocalTensor ROCAL_API_CALL rocalSequenceReaderSingleShard(RocalContext context,

128  const char* source_path,

129  RocalImageColor rocal_color_format,

130  unsigned shard_id,

131  unsigned shard_count,

132  unsigned sequence_length,

133  bool is_output,

134  bool shuffle = false,

135  bool loop = false,

136  unsigned step = 0,

137  unsigned stride = 0);

138 

153 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSource(RocalContext context,

154  const char* source_path,

155  const char* json_path,

156  RocalImageColor color_format,

157  unsigned internal_shard_count,

158  bool is_output,

159  bool shuffle = false,

160  bool loop = false,

161  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

162  unsigned max_width = 0, unsigned max_height = 0,

163  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

164 

181 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourcePartial(RocalContext p_context,

182  const char* source_path,

183  const char* json_path,

184  RocalImageColor rocal_color_format,

185  unsigned internal_shard_count,

186  bool is_output,

187  std::vector<float>& area_factor,

188  std::vector<float>& aspect_ratio,

189  unsigned num_attempts,

190  bool shuffle = false,

191  bool loop = false,

192  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

193  unsigned max_width = 0, unsigned max_height = 0);

194 

211 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourcePartialSingleShard(RocalContext p_context,

212  const char* source_path,

213  const char* json_path,

214  RocalImageColor rocal_color_format,

215  unsigned shard_id,

216  unsigned shard_count,

217  bool is_output,

218  std::vector<float>& area_factor,

219  std::vector<float>& aspect_ratio,

220  unsigned num_attempts,

221  bool shuffle = false,

222  bool loop = false,

223  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

224  unsigned max_width = 0, unsigned max_height = 0);

225 

241 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourceSingleShard(RocalContext context,

242  const char* source_path,

243  const char* json_path,

244  RocalImageColor color_format,

245  unsigned shard_id,

246  unsigned shard_count,

247  bool is_output,

248  bool shuffle = false,

249  bool loop = false,

250  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

251  unsigned max_width = 0, unsigned max_height = 0,

252  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

253 

268 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSource(RocalContext context,

269  const char* source_path,

270  RocalImageColor rocal_color_format,

271  unsigned internal_shard_count,

272  bool is_output,

273  bool shuffle = false,

274  bool loop = false,

275  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

276  unsigned max_width = 0, unsigned max_height = 0,

277  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

278 

295 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSourceSingleShard(RocalContext p_context,

296  const char* source_path,

297  RocalImageColor rocal_color_format,

298  unsigned shard_id,

299  unsigned shard_count,

300  bool is_output,

301  bool shuffle = false,

302  bool loop = false,

303  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

304  unsigned max_width = 0, unsigned max_height = 0,

305  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

306 

322 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSource(RocalContext context,

323  const char* source_path,

324  RocalImageColor rocal_color_format,

325  unsigned internal_shard_count,

326  bool is_output,

327  bool shuffle = false,

328  bool loop = false,

329  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

330  unsigned max_width = 0, unsigned max_height = 0,

331  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

332 

349 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSourceSingleShard(RocalContext p_context,

350  const char* source_path,

351  RocalImageColor rocal_color_format,

352  unsigned shard_id,

353  unsigned shard_count,

354  bool is_output,

355  bool shuffle = false,

356  bool loop = false,

357  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

358  unsigned max_width = 0, unsigned max_height = 0,

359  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

360 

376 extern "C" RocalTensor ROCAL_API_CALL rocalMXNetRecordSource(RocalContext context,

377  const char* source_path,

378  RocalImageColor rocal_color_format,

379  unsigned internal_shard_count,

380  bool is_output,

381  bool shuffle = false,

382  bool loop = false,

383  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

384  unsigned max_width = 0, unsigned max_height = 0,

385  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

386 

403 extern "C" RocalTensor ROCAL_API_CALL rocalMXNetRecordSourceSingleShard(RocalContext p_context,

404  const char* source_path,

405  RocalImageColor rocal_color_format,

406  unsigned shard_id,

407  unsigned shard_count,

408  bool is_output,

409  bool shuffle = false,

410  bool loop = false,

411  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

412  unsigned max_width = 0, unsigned max_height = 0,

413  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

414 

432 extern "C" RocalTensor ROCAL_API_CALL rocalFusedJpegCrop(RocalContext context,

433  const char* source_path,

434  RocalImageColor rocal_color_format,

435  unsigned num_threads,

436  bool is_output,

437  std::vector<float>& area_factor,

438  std::vector<float>& aspect_ratio,

439  unsigned num_attempts,

440  bool shuffle = false,

441  bool loop = false,

442  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

443  unsigned max_width = 0, unsigned max_height = 0);

444 

461 extern "C" RocalTensor ROCAL_API_CALL rocalFusedJpegCropSingleShard(RocalContext context,

462  const char* source_path,

463  RocalImageColor color_format,

464  unsigned shard_id,

465  unsigned shard_count,

466  bool is_output,

467  std::vector<float>& area_factor,

468  std::vector<float>& aspect_ratio,

469  unsigned num_attempts,

470  bool shuffle = false,

471  bool loop = false,

472  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

473  unsigned max_width = 0, unsigned max_height = 0);

474 

490 extern "C" RocalTensor ROCAL_API_CALL rocalJpegTFRecordSource(RocalContext context,

491  const char* source_path,

492  RocalImageColor rocal_color_format,

493  unsigned internal_shard_count,

494  bool is_output,

495  const char* user_key_for_encoded,

496  const char* user_key_for_filename,

497  bool shuffle = false,

498  bool loop = false,

499  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

500  unsigned max_width = 0, unsigned max_height = 0,

501  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

502 

519 extern "C" RocalTensor ROCAL_API_CALL rocalJpegTFRecordSourceSingleShard(RocalContext context,

520  const char* source_path,

521  RocalImageColor rocal_color_format,

522  unsigned shard_id,

523  unsigned shard_count,

524  bool is_output,

525  bool shuffle = false,

526  bool loop = false,

527  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

528  unsigned max_width = 0, unsigned max_height = 0,

529  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG);

530 

543 extern "C" RocalTensor ROCAL_API_CALL rocalRawTFRecordSource(RocalContext p_context,

544  const char* source_path,

545  const char* user_key_for_raw,

546  const char* user_key_for_filename,

547  RocalImageColor rocal_color_format,

548  bool is_output,

549  bool shuffle = false,

550  bool loop = false,

551  unsigned out_width = 0, unsigned out_height = 0,

552  const char* record_name_prefix = "");

553 

568 extern "C" RocalTensor ROCAL_API_CALL rocalRawTFRecordSourceSingleShard(RocalContext p_context,

569  const char* source_path,

570  RocalImageColor rocal_color_format,

571  unsigned shard_id,

572  unsigned shard_count,

573  bool is_output,

574  bool shuffle = false,

575  bool loop = false,

576  unsigned out_width = 0, unsigned out_height = 0,

577  const char* record_name_prefix = "");

578 

596 extern "C" RocalTensor ROCAL_API_CALL rocalVideoFileSource(RocalContext context,

597  const char* source_path,

598  RocalImageColor color_format,

599  RocalDecodeDevice rocal_decode_device,

600  unsigned internal_shard_count,

601  unsigned sequence_length,

602  bool is_output = false,

603  bool shuffle = false,

604  bool loop = false,

605  unsigned step = 0,

606  unsigned stride = 0,

607  bool file_list_frame_num = true);

608 

626 extern "C" RocalTensor ROCAL_API_CALL rocalVideoFileSourceSingleShard(RocalContext context,

627  const char* source_path,

628  RocalImageColor color_format,

629  RocalDecodeDevice rocal_decode_device,

630  unsigned shard_id,

631  unsigned shard_count,

632  unsigned sequence_length,

633  bool shuffle = false,

634  bool is_output = false,

635  bool loop = false,

636  unsigned step = 0,

637  unsigned stride = 0,

638  bool file_list_frame_num = true);

639 

658 extern "C" RocalTensor ROCAL_API_CALL rocalVideoFileResize(RocalContext context,

659  const char* source_path,

660  RocalImageColor color_format,

661  RocalDecodeDevice rocal_decode_device,

662  unsigned internal_shard_count,

663  unsigned sequence_length,

664  unsigned dest_width,

665  unsigned dest_height,

666  bool shuffle = false,

667  bool is_output = false,

668  bool loop = false,

669  unsigned step = 0,

670  unsigned stride = 0,

671  bool file_list_frame_num = true,

672  RocalResizeScalingMode scaling_mode = ROCAL_SCALING_MODE_DEFAULT,

673  std::vector<unsigned> max_size = {},

674  unsigned resize_shorter = 0,

675  unsigned resize_longer = 0,

676  RocalResizeInterpolationType interpolation_type = ROCAL_LINEAR_INTERPOLATION);

677 

697 extern "C" RocalTensor ROCAL_API_CALL rocalVideoFileResizeSingleShard(RocalContext context,

698  const char* source_path,

699  RocalImageColor color_format,

700  RocalDecodeDevice rocal_decode_device,

701  unsigned shard_id,

702  unsigned shard_count,

703  unsigned sequence_length,

704  unsigned dest_width,

705  unsigned dest_height,

706  bool shuffle = false,

707  bool is_output = false,

708  bool loop = false,

709  unsigned step = 0,

710  unsigned stride = 0,

711  bool file_list_frame_num = true,

712  RocalResizeScalingMode scaling_mode = ROCAL_SCALING_MODE_DEFAULT,

713  std::vector<unsigned> max_size = {},

714  unsigned resize_shorter = 0,

715  unsigned resize_longer = 0,

716  RocalResizeInterpolationType interpolation_type = ROCAL_LINEAR_INTERPOLATION);

717 

729 extern "C" RocalTensor ROCAL_API_CALL rocalRawCIFAR10Source(RocalContext context,

730  const char* source_path,

731  RocalImageColor color_format,

732  bool is_output,

733  unsigned out_width, unsigned out_height, const char* filename_prefix = "",

734  bool loop = false);

735 

741 extern "C" RocalStatus ROCAL_API_CALL rocalResetLoaders(RocalContext context);

742 

761 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSourcePartialSingleShard(RocalContext p_context,

762  const char* source_path,

763  RocalImageColor rocal_color_format,

764  unsigned shard_id,

765  unsigned shard_count,

766  bool is_output,

767  std::vector<float>& area_factor,

768  std::vector<float>& aspect_ratio,

769  unsigned num_attempts,

770  bool shuffle = false,

771  bool loop = false,

772  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

773  unsigned max_width = 0, unsigned max_height = 0);

774 

790 extern "C" RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSourcePartialSingleShard(RocalContext p_context,

791  const char* source_path,

792  RocalImageColor rocal_color_format,

793  unsigned shard_id,

794  unsigned shard_count,

795  bool is_output,

796  std::vector<float>& area_factor,

797  std::vector<float>& aspect_ratio,

798  unsigned num_attempts,

799  bool shuffle = false,

800  bool loop = false,

801  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

802  unsigned max_width = 0, unsigned max_height = 0);

817 extern "C" RocalTensor ROCAL_API_CALL rocalJpegExternalFileSource(RocalContext p_context,

818  RocalImageColor rocal_color_format,

819  bool is_output = false,

820  bool shuffle = false,

821  bool loop = false,

822  RocalImageSizeEvaluationPolicy decode_size_policy = ROCAL_USE_MOST_FREQUENT_SIZE,

823  unsigned max_width = 0, unsigned max_height = 0,

824  RocalDecoderType rocal_decoder_type = RocalDecoderType::ROCAL_DECODER_TJPEG,

825  RocalExternalSourceMode external_source_mode = RocalExternalSourceMode::ROCAL_EXTSOURCE_FNAME);

826 

838 extern "C" RocalTensor ROCAL_API_CALL rocalAudioFileSource(RocalContext context,

839  const char* source_path,

840  unsigned shard_count,

841  bool is_output,

842  bool shuffle = false,

843  bool loop = false,

844  bool downmix = false);

845 

858 extern "C" RocalTensor ROCAL_API_CALL rocalAudioFileSourceSingleShard(RocalContext p_context,

859  const char* source_path,

860  unsigned shard_id,

861  unsigned shard_count,

862  bool is_output,

863  bool shuffle = false,

864  bool loop = false,

865  bool downmix = false);

866 

867 #endif // MIVISIONX_ROCAL_API_DATA_LOADERS_H

rocalJpegCaffeLMDBRecordSource

RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSource(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for Caffe LMDB records. It allocates the resources and objects ...

rocalJpegFileSourceSingleShard

RocalTensor ROCAL_API_CALL rocalJpegFileSourceSingleShard(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder. It allocates the resources and objects required to read and de...

rocalVideoFileResize

RocalTensor ROCAL_API_CALL rocalVideoFileResize(RocalContext context, const char *source_path, RocalImageColor color_format, RocalDecodeDevice rocal_decode_device, unsigned internal_shard_count, unsigned sequence_length, unsigned dest_width, unsigned dest_height, bool shuffle=false, bool is_output=false, bool loop=false, unsigned step=0, unsigned stride=0, bool file_list_frame_num=true, RocalResizeScalingMode scaling_mode=ROCAL_SCALING_MODE_DEFAULT, std::vector< unsigned > max_size={}, unsigned resize_shorter=0, unsigned resize_longer=0, RocalResizeInterpolationType interpolation_type=ROCAL_LINEAR_INTERPOLATION)

Creates a video reader and decoder as a source. It allocates the resources and objects required to re...

rocalJpegExternalFileSource

RocalTensor ROCAL_API_CALL rocalJpegExternalFileSource(RocalContext p_context, RocalImageColor rocal_color_format, bool is_output=false, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG, RocalExternalSourceMode external_source_mode=RocalExternalSourceMode::ROCAL_EXTSOURCE_FNAME)

Creates JPEG external source image reader.

rocalFusedJpegCrop

RocalTensor ROCAL_API_CALL rocalFusedJpegCrop(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned num_threads, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

Creates JPEG image reader and partial decoder. It allocates the resources and objects required to rea...

rocalMXNetRecordSourceSingleShard

RocalTensor ROCAL_API_CALL rocalMXNetRecordSourceSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for MXNet records. It allocates the resources and objects requi...

rocalRawTFRecordSourceSingleShard

RocalTensor ROCAL_API_CALL rocalRawTFRecordSourceSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, unsigned out_width=0, unsigned out_height=0, const char *record_name_prefix="")

Creates Raw image loader. It allocates the resources and objects required to load images stored on th...

rocalSequenceReader

RocalTensor ROCAL_API_CALL rocalSequenceReader(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, unsigned sequence_length, bool is_output, bool shuffle=false, bool loop=false, unsigned step=0, unsigned stride=0)

Creates JPEG image reader and decoder. Reads [Frames] sequences from a directory representing a colle...

rocalJpegTFRecordSource

RocalTensor ROCAL_API_CALL rocalJpegTFRecordSource(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, const char *user_key_for_encoded, const char *user_key_for_filename, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates TensorFlow records JPEG image reader and decoder. It allocates the resources and objects requ...

rocalJpegFileSource

RocalTensor ROCAL_API_CALL rocalJpegFileSource(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder. It allocates the resources and objects required to read and de...

rocalJpegCOCOFileSourceSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourceSingleShard(RocalContext context, const char *source_path, const char *json_path, RocalImageColor color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader. It allocates the resources and objects required to read and decode COCO Jp...

rocalJpegCOCOFileSource

RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSource(RocalContext context, const char *source_path, const char *json_path, RocalImageColor color_format, unsigned internal_shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

JPEG image reader and decoder. It allocates the resources and objects required to read and decode COC...

rocalJpegTFRecordSourceSingleShard

RocalTensor ROCAL_API_CALL rocalJpegTFRecordSourceSingleShard(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates TensorFlow records JPEG image reader and decoder. It allocates the resources and objects requ...

rocalRawCIFAR10Source

RocalTensor ROCAL_API_CALL rocalRawCIFAR10Source(RocalContext context, const char *source_path, RocalImageColor color_format, bool is_output, unsigned out_width, unsigned out_height, const char *filename_prefix="", bool loop=false)

Creates CIFAR10 raw data reader and loader. It allocates the resources and objects required to read r...

rocalVideoFileSource

RocalTensor ROCAL_API_CALL rocalVideoFileSource(RocalContext context, const char *source_path, RocalImageColor color_format, RocalDecodeDevice rocal_decode_device, unsigned internal_shard_count, unsigned sequence_length, bool is_output=false, bool shuffle=false, bool loop=false, unsigned step=0, unsigned stride=0, bool file_list_frame_num=true)

Creates a video reader and decoder as a source. It allocates the resources and objects required to re...

rocalVideoFileSourceSingleShard

RocalTensor ROCAL_API_CALL rocalVideoFileSourceSingleShard(RocalContext context, const char *source_path, RocalImageColor color_format, RocalDecodeDevice rocal_decode_device, unsigned shard_id, unsigned shard_count, unsigned sequence_length, bool shuffle=false, bool is_output=false, bool loop=false, unsigned step=0, unsigned stride=0, bool file_list_frame_num=true)

Creates a video reader and decoder as a source. It allocates the resources and objects required to re...

rocalJpegCaffeLMDBRecordSourceSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSourceSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for Caffe LMDB records. It allocates the resources and objects ...

rocalMXNetRecordSource

RocalTensor ROCAL_API_CALL rocalMXNetRecordSource(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for MXNet records. It allocates the resources and objects requi...

rocalJpegCOCOFileSourcePartial

RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourcePartial(RocalContext p_context, const char *source_path, const char *json_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

JPEG image reader and partial decoder. It allocates the resources and objects required to read and de...

rocalJpegCaffe2LMDBRecordSource

RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSource(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned internal_shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for Caffe2 LMDB records. It allocates the resources and objects...

rocalVideoFileResizeSingleShard

RocalTensor ROCAL_API_CALL rocalVideoFileResizeSingleShard(RocalContext context, const char *source_path, RocalImageColor color_format, RocalDecodeDevice rocal_decode_device, unsigned shard_id, unsigned shard_count, unsigned sequence_length, unsigned dest_width, unsigned dest_height, bool shuffle=false, bool is_output=false, bool loop=false, unsigned step=0, unsigned stride=0, bool file_list_frame_num=true, RocalResizeScalingMode scaling_mode=ROCAL_SCALING_MODE_DEFAULT, std::vector< unsigned > max_size={}, unsigned resize_shorter=0, unsigned resize_longer=0, RocalResizeInterpolationType interpolation_type=ROCAL_LINEAR_INTERPOLATION)

Creates a video reader and decoder as a source. It allocates the resources and objects required to re...

rocalRawTFRecordSource

RocalTensor ROCAL_API_CALL rocalRawTFRecordSource(RocalContext p_context, const char *source_path, const char *user_key_for_raw, const char *user_key_for_filename, RocalImageColor rocal_color_format, bool is_output, bool shuffle=false, bool loop=false, unsigned out_width=0, unsigned out_height=0, const char *record_name_prefix="")

Creates Raw image loader. It allocates the resources and objects required to load images stored on th...

rocalJpegCaffeLMDBRecordSourcePartialSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCaffeLMDBRecordSourcePartialSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

Creates JPEG image reader and partial decoder for Caffe LMDB records. It allocates the resources and ...

rocalResetLoaders

RocalStatus ROCAL_API_CALL rocalResetLoaders(RocalContext context)

reset Loaders

rocalJpegCaffe2LMDBRecordSourceSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSourceSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0, RocalDecoderType rocal_decoder_type=RocalDecoderType::ROCAL_DECODER_TJPEG)

Creates JPEG image reader and decoder for Caffe2 LMDB records. It allocates the resources and objects...

rocalJpegCaffe2LMDBRecordSourcePartialSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCaffe2LMDBRecordSourcePartialSingleShard(RocalContext p_context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

Creates JPEG image reader and partial decoder for Caffe2 LMDB records. It allocates the resources and...

rocalJpegCOCOFileSourcePartialSingleShard

RocalTensor ROCAL_API_CALL rocalJpegCOCOFileSourcePartialSingleShard(RocalContext p_context, const char *source_path, const char *json_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

Creates JPEG image reader and partial decoder. It allocates the resources and objects required to rea...

rocalSequenceReaderSingleShard

RocalTensor ROCAL_API_CALL rocalSequenceReaderSingleShard(RocalContext context, const char *source_path, RocalImageColor rocal_color_format, unsigned shard_id, unsigned shard_count, unsigned sequence_length, bool is_output, bool shuffle=false, bool loop=false, unsigned step=0, unsigned stride=0)

Creates JPEG image reader and decoder. Reads [Frames] sequences from a directory representing a colle...

rocalFusedJpegCropSingleShard

RocalTensor ROCAL_API_CALL rocalFusedJpegCropSingleShard(RocalContext context, const char *source_path, RocalImageColor color_format, unsigned shard_id, unsigned shard_count, bool is_output, std::vector< float > &area_factor, std::vector< float > &aspect_ratio, unsigned num_attempts, bool shuffle=false, bool loop=false, RocalImageSizeEvaluationPolicy decode_size_policy=ROCAL_USE_MOST_FREQUENT_SIZE, unsigned max_width=0, unsigned max_height=0)

Creates JPEG image reader and partial decoder. It allocates the resources and objects required to rea...

RocalDecoderType

RocalDecoderType

rocAL Decoder Type enum

Definition: rocal_api_types.h:247

RocalResizeScalingMode

RocalResizeScalingMode

rocAL Resize Scaling Mode enum

Definition: rocal_api_types.h:298

RocalStatus

RocalStatus

rocAL Status enum

Definition: rocal_api_types.h:112

RocalImageSizeEvaluationPolicy

RocalImageSizeEvaluationPolicy

rocAL Image Size Evaluation Policy enum

Definition: rocal_api_types.h:175

RocalExternalSourceMode

RocalExternalSourceMode

RocalExternalSourceMode struct.

Definition: rocal_api_types.h:368

RocalDecodeDevice

RocalDecodeDevice

rocAL Decode Device enum

Definition: rocal_api_types.h:196

RocalContext

void * RocalContext

typedef void* rocAL Context

Definition: rocal_api_types.h:63

RocalResizeInterpolationType

RocalResizeInterpolationType

rocAL Resize Interpolation Type enum

Definition: rocal_api_types.h:320

RocalImageColor

RocalImageColor

rocAL Image Color enum

Definition: rocal_api_types.h:133

ROCAL_DECODER_TJPEG

@ ROCAL_DECODER_TJPEG

AMD ROCAL_DECODER_TJPEG.

Definition: rocal_api_types.h:250

ROCAL_SCALING_MODE_DEFAULT

@ ROCAL_SCALING_MODE_DEFAULT

scales wrt specified size, if only resize width/height is provided the other dimension is scaled acco...

Definition: rocal_api_types.h:301

ROCAL_USE_MOST_FREQUENT_SIZE

@ ROCAL_USE_MOST_FREQUENT_SIZE

AMD ROCAL_USE_MOST_FREQUENT_SIZE.

Definition: rocal_api_types.h:184

ROCAL_EXTSOURCE_FNAME

@ ROCAL_EXTSOURCE_FNAME

list of filename passed as input

Definition: rocal_api_types.h:371

ROCAL_LINEAR_INTERPOLATION

@ ROCAL_LINEAR_INTERPOLATION

AMD ROCAL_LINEAR_INTERPOLATION.

Definition: rocal_api_types.h:326

rocalAudioFileSourceSingleShard

RocalTensor ROCAL_API_CALL rocalAudioFileSourceSingleShard(RocalContext p_context, const char *source_path, unsigned shard_id, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, bool downmix=false)

rocalAudioFileSource

RocalTensor ROCAL_API_CALL rocalAudioFileSource(RocalContext context, const char *source_path, unsigned shard_count, bool is_output, bool shuffle=false, bool loop=false, bool downmix=false)

rocal_api_types.h

The AMD rocAL Library - Types.

ROCAL_API_CALL

#define ROCAL_API_CALL

Definition: rocal_api_types.h:35