add tool: tfrecord to mindrecord

5 years ago · 3353a20d8b
parent cb706951c1
commit 3353a20d8b
4 changed files with 671 additions and 1 deletions
--- a/mindspore/mindrecord/init.py
+++ b/mindspore/mindrecord/init.py
@ -31,7 +31,8 @@ from .tools.cifar10_to_mr import Cifar10ToMR
 from .tools.cifar100_to_mr import Cifar100ToMR
 from .tools.imagenet_to_mr import ImageNetToMR
 from .tools.mnist_to_mr import MnistToMR
 from .tools.tfrecord_to_mr import TFRecordToMR
 __all__ = ['FileWriter', 'FileReader', 'MindPage',
-           'Cifar10ToMR', 'Cifar100ToMR', 'ImageNetToMR', 'MnistToMR',
+           'Cifar10ToMR', 'Cifar100ToMR', 'ImageNetToMR', 'MnistToMR', 'TFRecordToMR',
           'SUCCESS', 'FAILED']
--- a/mindspore/mindrecord/tools/tfrecord_to_mr.py
+++ b/mindspore/mindrecord/tools/tfrecord_to_mr.py
--- a/tests/ut/data/mindrecord/testTFRecordData/README.md
+++ b/tests/ut/data/mindrecord/testTFRecordData/README.md
@ -0,0 +1 @@
 ## tfrecord file dir
--- a/tests/ut/python/mindrecord/test_tfrecord_to_mr.py
+++ b/tests/ut/python/mindrecord/test_tfrecord_to_mr.py