图像分类 Pytorch答案

【问题标题】：Image Classification Pytorch图像分类 Pytorch
【发布时间】：2018-12-02 05:23:03
【问题描述】：

如何确定 CNN 中的层数和最终模型以提高预测的准确性。我正在对图像进行分类，目前使用简单模型获得 65% 的准确度，我应该如何增强它以实现最大准确度。（火炬）

【问题讨论】：

【解决方案1】：

我想说三件事。

1) 检查 torchvision.models link 在那里你可以找到预训练的优秀模型，如果你将层设置为不需要渐变并且只修改最后一层以具有正确数量的类，它们将为你提供出色的性能

2) 在加载图像时玩转变换link 这可以帮助你

3) 使用最后一层的数量、不同的优化器并尝试调度器link（这将在训练期间调整学习率以获得更好的拟合）

希望对你有帮助:)

【讨论】：