l2181265的博客

c++测试pytorch训练的模型pytorch训练的pth模型转换成onnx模型，使用c++测试onnx模型。1. 模型.pth转.onnx化繁为简写那么多废话不如简单明了import torchfrom Unet import Unetdef pth2onnx(input, pth_path, onnx_path): model = Unet() # 导入自己的网络模型 model.load_state_dict(torch.load(pth_path))

1、下载anaconda3.5.0.1------python3.6.3版本，3.6.0版本调用pytorch会出问题。。anaconda下载地址2、配置pytorch1.4.0 & torchvision0.5.0。。pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/si

np.reshape增加一维举例import numpy as np# 数组img, (10, 10)img = np.arange(100).reshape(10, 10)print(img)[[ 0 1 2 3 4 5 6 7 8 9] [10 11 12 13 14 15 16 17 18 19] [20 21 22 23 24 25 26 27 28 29] [30 31 32 33 34 35 36 37 38 39] [40 41 42 43 44 45

参考：添加链接描述

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 6 and 8 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:612数据纬度不对造成的。

定义网络：import torch.nn as nnimport torch.nn.functional as Fclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.con...