!2135 add tfrecord to mindrecord tool

Merge pull request !2135 from guozhijian/add_tools_tfrecord_to_mindrecord
5 years ago · 2fa2f86619
parent c518391652 3353a20d8b
commit 2fa2f86619
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