以下是关于如何对其进行微小修改的指南 MNIST的例子 满足您的需求。可能有一些摆弄。
train/cats/abc.jpg train/cats/def.jpg ... train/dogs/ghi.jpg ... train/mouse/jkl.jpg ... analogously for validation val/cats/... val/dogs/...
文件名无关紧要,只是这些目录将用于识别类别。
train_data = mx.gluon.data.vision.datasets.ImageFolderDataset(training_path) val_data = mx.gluon.data.vision.datasets.ImageFolderDataset(val_path)
请注意,您可能需要对图像应用变换。
net = Dense(number_of_outputs) # activation=None, use_bias=True by default
(在代码中 lenet 代替 net 。)
lenet
net
更多文档: https://mxnet.incubator.apache.org/versions/master/tutorials/gluon/datasets.html
希望有所帮助!