小编mja*_*kie的帖子

如何在 PyTorch 中更改图片大小

我试图转换CNN Keras模型使用FER2013数据集PyTorch模型情感识别和我有以下错误:

Traceback (most recent call last):
  File "VGG.py", line 112, in <module>
    transfer.keras_to_pytorch(keras_network, pytorch_network)
  File "/home/eorg/NeuralNetworks/user/Project/model/nntransfer.py", line 121, in keras_to_pytorch
    pytorch_model.load_state_dict(state_dict)
  File "/home/eorg/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 334, in load_state_dict
    own_state[name].copy_(param)
RuntimeError: inconsistent tensor size at /b/wheel/pytorch-src/torch/lib/TH/generic/THTensorCopy.c:51
Run Code Online (Sandbox Code Playgroud)

我知道错误与图像的形状有关。在 Keras 中,输入大小定义为 48 x 48。

我的问题是如何在 PyTorch 模型中定义我的图片是 48x48 的形状?我在文档和示例中找不到这样的功能。

任何帮助都会有用!

conv-neural-network pytorch

7
推荐指数
1
解决办法
2万
查看次数

标签 统计

conv-neural-network ×1

pytorch ×1