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实现检测服务器是否可以ping通的2种方法
Jan 01 Python
Python在Windows和在Linux下调用动态链接库的教程
Aug 18 Python
十个Python程序员易犯的错误
Dec 15 Python
Python语言实现获取主机名根据端口杀死进程
Mar 31 Python
Python爬虫实例爬取网站搞笑段子
Nov 08 Python
用不到50行的Python代码构建最小的区块链
Nov 16 Python
对python以16进制打印字节数组的方法详解
Jan 24 Python
提升Python程序性能的7个习惯
Apr 14 Python
python调用webservice接口的实现
Jul 12 Python
VScode连接远程服务器上的jupyter notebook的实现
Apr 23 Python
浅谈django 重载str 方法
May 19 Python
python中有函数重载吗
May 28 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
Zerg基本策略
2020/03/14 星际争霸
php处理restful请求的路由类分享
2014/02/27 PHP
用JavaScript事件串连执行多个处理过程的方法
2007/03/09 Javascript
jquery简单实现滚动条下拉DIV固定在头部不动
2013/11/25 Javascript
简单的js图片轮换代码(js图片轮播)
2014/05/06 Javascript
将JavaScript的jQuery库中表单转化为JSON对象的方法
2015/11/17 Javascript
jQuery插件Validate实现自定义表单验证
2016/01/18 Javascript
JS获取html元素的标记名实现方法
2016/10/08 Javascript
随机生成10个不重复的0-100的数字(实例讲解)
2017/08/16 Javascript
微信小程序radio组件使用详解
2018/01/31 Javascript
Vue实现表格中对数据进行转换、处理的方法
2018/09/06 Javascript
详解@angular/cli 改变默认启动端口两种方式
2018/11/29 Javascript
vuex+axios+element-ui实现页面请求loading操作示例
2020/02/02 Javascript
基于JS+HTML实现弹窗提示是否确认提交功能
2020/06/17 Javascript
[01:37]DOTA2超级联赛专访ChuaN 传奇般的电竞之路
2013/06/19 DOTA
Python基础学习之时间转换函数用法详解
2019/06/18 Python
详解python 利用echarts画地图(热力图)(世界地图,省市地图,区县地图)
2019/08/06 Python
Python各种扩展名区别点整理
2020/02/27 Python
Python Socket TCP双端聊天功能实现过程详解
2020/06/15 Python
python3从网络摄像机解析mjpeg http流的示例
2020/11/13 Python
AmazeUI中模态框的实现
2020/08/19 HTML / CSS
荷兰男士时尚网上商店:Suitable
2017/12/25 全球购物
this关键字的作用
2016/01/30 面试题
岗位职责定义及内容
2013/11/08 职场文书
学校招生宣传广告词
2014/03/19 职场文书
《明天,我们毕业》教学反思
2014/04/24 职场文书
我的小天地教学反思
2014/04/30 职场文书
授权委托书(完整版)
2014/09/10 职场文书
销售经理岗位职责范本
2015/04/02 职场文书
浅谈Python列表嵌套字典转化的问题
2021/04/07 Python
详解Laravel服务容器的优势
2021/05/29 PHP
python中的class_static的@classmethod的巧妙用法
2021/06/22 Python
python数据可视化JupyterLab实用扩展程序Mito
2021/11/20 Python
Innodb存储引擎中的后台线程详解
2022/04/03 MySQL
Docker部署Mysql8的实现步骤
2022/07/07 Servers
Java使用HttpClient实现文件下载
2022/08/14 Java/Android