2024年10月24日
深度学习各种错误汇总
摘要： 1、RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 如上错误是在 torch.onnx.export 时出现的 解决：x造成的，后边加上. 阅读全文
posted @ 2024-10-24 18:18 夕西行 阅读(43) 评论(0) 推荐(0)
样本预处理
摘要： 以500*332的图为例 import torchvision.transforms as transforms from PIL import Image img = Image.open('D:/cat.jpg') #缩放成高宽都是256的正方形（小变大的话，是插值不会填充黑边） resize 阅读全文
posted @ 2024-10-24 16:40 夕西行 阅读(26) 评论(0) 推荐(0)