silentkunden的博客

''' Resnet18 代码 date:2019/11/30 author:hzp'''import torchimport torch.nn as nnimport torch.nn.functional as Fclass ResidualBlock(nn.Module): def __init__(self, inchannel, outch...

图像处理：先对所有的图像的大小reshape到224*224（Resnet输入为224*224）def Image_PreProcessing(imagepath, targetpath): # 待处理图片存储路径 im = cv2.imread(imagepath, 1) h, w, _ = im.shape print(im) t = 0 ...

损失函数，是编译一个神经网络模型必须的两个参数之一，另一个是优化器。损失函数是指用于计算标签值和预测值之间差异的函数，常见的有多种损失函数可供选择，典型的有距离向量，绝对值向量等。 nn.L1Loss L1Loss 计算方法比较简单，原理就是取预测值和真实值的绝对误差的平均数。计算公式如下 ...