加載NORB數據集與Python中的Keras Tensorflow一起使用

Question

我想用NORB數據集（http://www.cs.nyu.edu/~ylclab/data/norb-v1.0-small/）做一些工作，但是我無法從包含數據集的二進制文件中讀取它。任何身體都可以幫忙？

我試過numpy.fromfile()，但它以錯誤的方式讀取文件a。

Answer 1

我最近有同樣的問題，因爲我不得不使用該數據集，我發現它以一種奇怪的二進制格式分發。

爲此，我製作了一個python包裝，您可能會覺得有用。你可以找到它here。

使用很簡單，只要：

dataset = SmallNORBDataset(dataset_root='small_norb_root') 

其中small_norb_root是在數據集中的文件已被下載並解壓縮的文件夾。所有解析的東西都在類的初始化過程中發生，並且對用戶來說是透明的。一旦初始化，您可以分別找到所需的數據：dataset.data['train']或dataset.data['test']。

也可以探索與數據集：

dataset.explore_random_examples(dataset_split='train') 

如果一切順利，輸出應如下所示：