Add ofrecord_image_classification_reader (#3419)
* ofrecord_image_classification_reader * refine * use num_decode_threads_per_machine * Specify image_feature_name and label_feature_name * Extract LoadWorker and DecodeWorker * Remove image_feature_name/label_feature_name default value * underlaying_=>base * set decode_buffer_size_per_thread to 32
Showing
- oneflow/python/ops/user_data_ops.py 46 additions, 0 deletionsoneflow/python/ops/user_data_ops.py
- oneflow/python/test/models/alexnet.py 4 additions, 5 deletionsoneflow/python/test/models/alexnet.py
- oneflow/user/data/ofrecord_image_classification_data_reader.h 58 additions, 0 deletions...low/user/data/ofrecord_image_classification_data_reader.h
- oneflow/user/data/ofrecord_image_classification_dataset.h 207 additions, 0 deletionsoneflow/user/data/ofrecord_image_classification_dataset.h
- oneflow/user/data/ofrecord_image_classification_parser.h 59 additions, 0 deletionsoneflow/user/data/ofrecord_image_classification_parser.h
- oneflow/user/kernels/ofrecord_image_classification_reader_kernel.cpp 62 additions, 0 deletions...r/kernels/ofrecord_image_classification_reader_kernel.cpp
- oneflow/user/ops/ofrecord_image_classification_reader_op.cpp 72 additions, 0 deletionsoneflow/user/ops/ofrecord_image_classification_reader_op.cpp
Please register or sign in to comment