查询到最新的3条

torchvision.utils.save_image()保存tensor显示图片异常问题解决

有些像素点会显示为全黑(灰度图),如下图所示,第一张和第三张图 刚开始以为是图像数据分布范围的问题,在保存之前输出图像tensor的最大max和最小min值,出现了 -0.0x和1.0x的数值,说明图像的像素范围超出了0-1。 可是通过读utils.save_image()的源码发现,就算超出0-1也不应该出现这种问题,源码中存在如下部分代码 # 继续阅读

onnxruntime推理

pytorch模型训练 这里以pytorch平台和mobilenet v2网络为例,给出模型的训练过程。具体代码如下所示: import os import torchvision.transforms as transforms from torchvision import datasets import torch.utils.data as data import torch import numpy as np import torchvision.mo 继续阅读

一文学会Anaconda!

我使用的pytoch版本: conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia 安装放后面了,参考https://blog.csdn.net/wq_ocean_/article/details/103889237 1. 使用conda进行pyt 继续阅读