构建网络遇到的问题-AlexNet
实质上train_dataset里面的内容是由所有图片进行排列的,每一行都由(img,类别号)组成。7.在模型训练完后,用图片去让模型进行分类,要注意模型使用的图片大小,输入的图像一般要是4维的。若当前文件是train.py位于在D:\projects\AlexNet\train.py。classes_to_idx:即目录下文件夹(类别)的对应的id,从0开始。8.模型最后输出一般是一个二维数组,(图片个数,类别个数)classes:即目录下文件夹(类别)的个数。transform=数据要进行的操作。
原创
2024-09-25 19:07:46 ·
816 阅读 ·
0 评论