[tools/dataset_converter] Add the mnist dataset (Samsung#12699)

This PR adds the mnist tensorflow dataset to the tf_dataset_converter. ONE-DCO-1.0-Signed-off-by: SeungHui Youn <[email protected]>
lemmaa · Feb 28, 2024 · ecd17dd · ecd17dd
1 parent 79ca151
commit ecd17dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/generate_datafile/tf_dataset_converter/datasets.py b/tools/generate_datafile/tf_dataset_converter/datasets.py
@@ -9,7 +9,7 @@
 
 class Mnist():
     def check(dataset_name):
-        supported_dataset_list = ['fashion_mnist']
+        supported_dataset_list = ['fashion_mnist', 'mnist']
         if dataset_name in supported_dataset_list:
             return
         print(f'{dataset_name} does not fit Mnist')