![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
数据读取
CV干饭王
这个作者很懒,什么都没留下…
展开
-
torchvision.transforms.ToTensor()与torchvision.transforms.ToPILImage()详解
torchvision.transforms.ToTensor()pytorch在加载数据集时都需要对数据记性transforms转换,其中最常用的就是torchvision.transforms.ToTensor()函数,但初学者可以认为这个函数只是把输入数据类型转换为pytorch的Tensor(int64)类型,其实不然,该函数内部的具体转换步骤为:1、将图片转化成内存中的存储格式;2、将字节以流的形式输入,转化成Tensor类型;3、对Tensor进行reshape;4、对Tensor进原创 2020-12-04 10:09:10 · 5318 阅读 · 0 评论 -
python利用文件夹下xml格式标签文件批量裁剪出图片中的目标(文件夹、图片名称、目标框数量无限制,逐行注释)
这段时间有个小任务是提取出图片中的手部物体,然后我就用labelimg打标软件给部分图片打了标(生成xml标签文件),顺手写了个小脚本将这些目标从原图中裁剪出来,效果如下所示,感觉还可以。原始图片(实例1):裁剪效果(实例1):原始图片(实例2):裁剪效果(实例2,此处空手为打标,故未裁剪):完整代码本代码针对以下问题均有解决:1、同一xml文件有多个目标框;2、图片和xml文件在不同或相同文件夹下;3、部分图片未打标,对应的xml文件不存在;4、图片文件夹中有非图片(png,原创 2020-12-03 10:37:15 · 2846 阅读 · 11 评论 -
pytorch加载自己的数据集,通过读取train.txt、test.txt文件(附数据集txt生成完整代码,注释详细)
pytorch读取指定train.txt、test.txt文件加载自己的数据集txt生成脚本完整代码如下:读取自己的数据集,打乱并划分,生成train.txt、test.txt (每一行为图片的绝对路径+标签,完整代码,注释详细)https://blog.csdn.net/weixin_44414948/article/details/110205546train.txt、test.txt示例如下图所示:先占坑,回去补图import torchfrom torchvision import原创 2020-11-26 21:48:48 · 13331 阅读 · 3 评论 -
python读取数据集文件下所有文件并打乱划分生成训练测试txt文件(生成train.txt、test.txt,顺序随机,默认比例8:2)
因为每次读取数据集都要写加载代码,十分浪费时间,而我本人又比较懒。故写了个python小工具,一键运行,读取数据集文件下所有文件并打乱划分生成训练测试txt文件(生成train.txt、test.txt,顺序随机,默认比例8:2),十分方便。运行脚本前需要将文件夹整理为这类的形式:root = “C:\Users\hq\Desktop\HoldingObject\pokemon”(记得改为自己的根路径)路径树如下(文件夹、图片数量无限制):—pokemon——0(此处文件夹名为标签)————1原创 2020-11-26 21:38:31 · 7568 阅读 · 6 评论