Pytorch
Heiseweiye
这个作者很懒,什么都没留下…
展开
-
Pytorch学习之transforms(三)
ToTensor()Convert a PIL Image or numpy.ndarray to tensor. This transform does not support torchscript.目前学到的一般的图片转换类型有如下三种PIL ImageImage.open('路径')numpy.ndarraycv.imread('路径')或np.array(PIL Image)torch.TensorToTensor(PIL Image or numpy.ndarray)下面代原创 2021-05-25 16:58:05 · 171 阅读 · 0 评论 -
Pytorch学习(二)
Pytorch tensorboard使用SummaryWriter模块实现数据可视化所用方法包括如下 - 实例化方法——SummaryWriter('目录') - Image对象——Image.open('路径') - nparray转换——np.array('Image对象') - 绘制图片——writer.add_image('标签', numpy.array对象, step, dataformats) - 坐标轴绘制——writer.add_scalar('标签', y轴, x轴)原创 2021-05-25 14:50:45 · 147 阅读 · 0 评论 -
Pytorch学习(一)
pytorch Dataset通过Dataset类自定义一个数据集类该自定义数据集类可以通过路径创建一个训练集通过图片名称创建Image对象from torch.utils.data import Datasetfrom PIL import Imageimport osclass Mydata(Dataset): ''' 自定义数据集类 ''' def __init__(self, root_dir, label_dir): '''原创 2021-05-25 14:41:36 · 92 阅读 · 0 评论