pytorch 实现将自己的图片数据处理成可以训练的图片类型


Posted in Python onJanuary 08, 2020

为了使用自己的图像数据,需要仿照pytorch数据输入创建新的类,其中数据格式为numpy.ndarray。

将自己的图片保存到numpy.ndarray中,然后创建类

from torch.utils.data import Dataset
import numpy as np
 
 
class Dataset(Dataset):
  def __init__(self, path_img, path_target, transforms=None):
    self.train = path_img
    self.targets = path_target
    self.transforms = transforms
 
  def __len__(self):
    return len(self.train)
 
  def __getitem__(self, idx):
    img = self.train[idx]
    target = self.targets[idx]
 
    if self.transforms:
      img = self.transforms(img)
      target = self.transforms(target)
 
    return img, target

使用方法和Mnist数据一样的使用方法

isbi = Dataset(imgs_train, imgs_mask_train,
            transforms=transform)
dataload=torch.utils.data.DataLoader(isbi,batch_size=4,shuffle=True)
for i, data in enumerate(dataload, 1):
  img,label=data
  print img.shape
  print img.shape
  print 10*'*'

以上这篇pytorch 实现将自己的图片数据处理成可以训练的图片类型就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python执行子进程实现进程间通信的方法
Jun 02 Python
python 网络爬虫初级实现代码
Feb 27 Python
python实现给微信公众号发送消息的方法
Jun 30 Python
django多个APP的urls设置方法(views重复问题解决)
Jul 19 Python
python多环境切换及pyenv使用过程详解
Sep 27 Python
window7下的python2.7版本和python3.5版本的opencv-python安装过程
Oct 24 Python
pytorch中的上采样以及各种反操作,求逆操作详解
Jan 03 Python
python-xpath获取html文档的部分内容
Mar 06 Python
django admin后管定制-显示字段的实例
Mar 11 Python
Tensorflow之MNIST CNN实现并保存、加载模型
Jun 17 Python
Pycharm自动添加文件头注释和函数注释参数的方法
Oct 23 Python
python爬不同图片分别保存在不同文件夹中的实现
Apr 02 Python
pytorch下大型数据集(大型图片)的导入方式
Jan 08 #Python
Python 实现训练集、测试集随机划分
Jan 08 #Python
Pyecharts绘制全球流向图的示例代码
Jan 08 #Python
PyTorch 解决Dataset和Dataloader遇到的问题
Jan 08 #Python
使用PyTorch将文件夹下的图片分为训练集和验证集实例
Jan 08 #Python
使用 PyTorch 实现 MLP 并在 MNIST 数据集上验证方式
Jan 08 #Python
Pycharm 2020最新永久激活码(附最新激活码和插件)
Sep 17 #Python
You might like
php 无限级分类学习参考之对ecshop无限级分类的解析 带详细注释
2010/03/23 PHP
ci检测是ajax还是页面post提交数据的方法
2014/11/10 PHP
8个必备的PHP功能开发
2015/10/02 PHP
PHP递归删除多维数组中的某个值
2017/04/17 PHP
PHP CURL中传递cookie的方法步骤
2019/05/09 PHP
JS获取IUSR_机器名和IWAM_机器名帐号的密码
2006/12/06 Javascript
List Information About the Binary Files Used by an Application
2007/06/18 Javascript
Javascript中 关于prototype属性实现继承的原理图
2013/04/16 Javascript
JS常用表单验证方法总结
2014/05/22 Javascript
jQuery 过滤方法filter()选择具有特殊属性的元素
2014/06/15 Javascript
JavaScript中定义函数的三种方法
2015/03/12 Javascript
通过伪协议解决父页面与iframe页面通信的问题
2015/04/05 Javascript
javascript学习小结之prototype
2015/12/03 Javascript
jquery实现轮播图效果
2017/02/13 Javascript
在Swiper内如何制作CSS3动画效果示例代码
2017/12/07 Javascript
AngularJS与后端php的数据交互方法
2018/08/13 Javascript
vue中$refs, $emit, $on, $once, $off的使用详解
2019/05/26 Javascript
vue使用i18n实现国际化的方法详解
2019/09/05 Javascript
layui 实现二级弹窗弹出之后 关闭一级弹窗的方法
2019/09/18 Javascript
详解小程序BackgroundAudioManager踩坑之旅
2019/12/08 Javascript
JS如何在数组指定位置插入元素
2020/03/10 Javascript
[01:17]辉夜杯战队访谈宣传片—EHOME
2015/12/25 DOTA
[02:49]DAC2018决赛日TOP5 LGD开启黑暗之门绝杀VP
2018/04/08 DOTA
python使用PyGame播放Midi和Mp3文件的方法
2015/04/24 Python
python绘制地震散点图
2019/06/18 Python
对DJango视图(views)和模版(templates)的使用详解
2019/07/17 Python
美国农场鲜花速递:The Bouqs
2018/07/13 全球购物
简历自荐信
2013/12/02 职场文书
后勤部长岗位职责
2013/12/14 职场文书
工商管理毕业生推荐信
2013/12/24 职场文书
电子银行营销方案
2014/02/22 职场文书
区级文明单位申报材料
2014/05/15 职场文书
2015年银行大堂经理工作总结
2015/04/24 职场文书
一篇文章带你复习java知识点
2021/06/28 Java/Android
SpringCloud之@FeignClient()注解的使用方式
2021/09/25 Java/Android
如何让你的Nginx支持分布式追踪详解
2022/07/07 Servers