怎么使用mxnet将自己的数据集图片做成mnist数据集
时间: 2023-07-09 10:38:49 浏览: 65
要使用mxnet将自己的数据集图片做成mnist数据集,可以按照以下步骤进行:
1. 准备自己的数据集,包括图片以及对应的标签信息。图片应该是灰度图,且尺寸应该是28x28。
2. 安装mxnet,并导入必要的模块:
```python
import mxnet as mx
import numpy as np
import os
```
3. 定义一个函数来加载图片和标签数据:
```python
def load_data(path):
with open(os.path.join(path, 'data.npy'), 'rb') as f:
data = np.load(f)
with open(os.path.join(path, 'label.npy'), 'rb') as f:
label = np.load(f)
return data, label
```
4. 定义一个函数来将图片和标签数据转换为mxnet的数据格式:
```python
def transform(data, label):
return mx.nd.array(data), mx.nd.array(label)
```
5. 加载自己的数据集,并将其转换为mxnet的数据格式:
```python
data, label = load_data('path/to/your/dataset')
data, label = transform(data, label)
```
6. 加载mnist数据集,并将其转换为mxnet的数据格式:
```python
train_data = mx.gluon.data.vision.MNIST(train=True)
test_data = mx.gluon.data.vision.MNIST(train=False)
train_data = train_data.transform_first(transform)
test_data = test_data.transform_first(transform)
```
7. 将自己的数据集与mnist数据集合并:
```python
train_data = mx.gluon.data.ConcatenatedDataset(train_data, mx.gluon.data.ArrayDataset(data, label))
```
8. 最后,可以像使用mnist数据集一样使用合并后的数据集来训练模型。
以上就是使用mxnet将自己的数据集图片做成mnist数据集的步骤,希望能对你有所帮助。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)