2301_76756147的博客

原创 【Pytorch图像分类】MobilenetV2手语识别（超详细1W+）

这个类会自动读取指定目录下的图像，并将它们分为不同的类别（所以目录结构很重要，见 “五、注意事项（一）目录结构”）是PyTorch的一个视觉库，它提供了很多视觉任务相关的工具，比如数据集加载器、模型和变换（transforms）。这个函数使用预处理后的图像作为输入，经过模型预测后，返回经过softmax处理的概率分布和预测类别的索引。通过这个函数，我们可以方便地加载数据，为后续的模型训练做好准备。最后，我们从训练数据集中获取类别名称，并将训练数据加载器、验证数据加载器和类别名称返回。