自用pytorch之图片、文本导入、标签导出（小土堆）
学习路线 爆肝杰哥 装一系列东西 在G盘
小土堆
首先插入 dataset： from torch.utils.data import Dataset
继承dateset类：
获取照片路径：
from PIL import Image
img path=‘’D//。。。。。。。。‘’ img_path是一个字符串变量，添加地址时注意 两个斜杠
img=Image.open(img_path) 变量img获取的是图片， 用的是Image.open的方法
获取一组照片：
导入路径：dir_path=“D:\PyCharm\Py_Projects\XiaoTuDui\dataset\train\ants” dir_path是一个字符串变量
import os os有方法
img_path_list=os.listdir(dir_path) 通过os的方法，导入文件夹路径从而创建列表
实战讲解
目录如下：
实战
→
up用的相对路径 root_dir=“dataset/train”
lable_dir=“ants”
两个路径 拼接功能： path=os.path.join(root_dir,lable_dir)
然后我来：root_dir复制的根路径
整合后 path=‘XiaoTuDui\datase\train\ants’
构建全局变量，相当于全局函数
检查时报错了。于是我换成了绝对地址
此时img_path成为图片的列表。
构建类（相当于一个新函数，idx是自变量）
定义img_name是地址中图片的名称（字符串类型） （通过列表img_path传输）
比如我让idx=0试了下
在这里插入代码片
```*斜体样式*
import os
from PIL import Image *PIL库可以插Image*
root_dir="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\dataset\\train" *这里最好导入绝对地址*
lable_dir="ants" *ants是train的下级文件，既是地址，也是标签*
path=os.path.join(root_dir,lable_dir) *os.path,join起到粘连左右，把地址输出，以字符串类型输出*
img_path=os.listdir(path) *通过os.listdir把照片排序，生成列表给 img_path*
img_name=img_path[idx] *把列表的第【idx】项传递给img_name* 效果如下
```python **
构造一个类。相当于构建一个获取项目的函数
def __getitem__(self, idx):
img_name=self.img_path[idx] //把全局变量中的照片名字传递给 img_name//
img_item_path=os.path.join(self.root_dir,self.lable_dir,img_name) //定位照片具体位置（最精确位置）//
img=Image.open(img_item_path) //把地址中的图片打开传递给img//
lable=self.lable_dir //把本次ants文件夹的标签传递给lable//
return img,lable //结果返回为照片，标签
def __len__(self):
return len(self.img_path)
计算长度。
完整体的dataset类如下（相当于一个完整的函数）
class MyData(Dataset):
def __init__(self,root_dir,lable_dir):
self.root_dir=root_dir
self.lable_dir=lable_dir
self.path=os.path.join(root_dir,lable_dir)
self.img_path=os.listdir(self.path)
def __getitem__(self, idx):
img_name=self.img_path[idx]
img_item_path=os.path.join(self.root_dir,self.lable_dir,img_name)
img=Image.open(img_item_path)
lable=self.lable_dir
return img,lable
def __len__(self):
return len(self.img_path)
按时间顺序向下，
root_dir="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\dataset\\train"
ants_lable_dir="ants"
ants_dataset=MyData(root_dir,ants_lable_dir)
点此处重置
然后在控制台依次输入
1、
from torch.utils.data import Dataset
from PIL import Image
import os
class MyData(Dataset):
def __init__(self,root_dir,lable_dir):
self.root_dir=root_dir
self.lable_dir=lable_dir
self.path=os.path.join(root_dir,lable_dir)
self.img_path=os.listdir(self.path)
def __getitem__(self, idx):
img_name=self.img_path[idx]
img_item_path=os.path.join(self.root_dir,self.lable_dir,img_name)
img=Image.open(img_item_path)
lable=self.lable_dir
return img,lable
def __len__(self):
return len(self.img_path)
2、
root_dir="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\dataset\\train"
ants_lable_dir="ants"
3、
ants_dataset=MyData(root_dir,ants_lable_dir)
右侧结果如图似乎所有照片都以列表形式传入ants_dataset
可以看到root_dir
path等
4、
如果我们要第一个变量可以输入
ants_dataset[0]
5、
输入：
img,label=ants_dataset[0]
img.show()
由于我们在My_Data类中创建了
所以可以返回 img lable
6、然后我们可以添加两行代码，传入密封的数据集
控制台输入
root_dir="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\dataset\\train"
ants_lable_dir="ants"
bees_lable_dir="bees"
ants_dataset=MyData(root_dir,ants_lable_dir)
bees_dataset=MyData(root_dir,bees_lable_dir)
img,lable=bees_dataset[1]
img,show（）
则出现蜜蜂了
进阶，如果数据集很多
train_dataset=ants_dataset+bees_dataset
右侧可以看到被整合到一起了
1、具体查看
发现总数据集长度245，其中蚂蚁124.。。我们把第123传给img，lable，发现img是蚂蚁。
加深印象，再进阶
新建一个文件夹
把之前的ants改成ants——image 同时新建lable文件夹，在lable中，创建txt
输入代码
import os
root_dir ="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\datasets_jinjieP7\\train" //设置根目录//
target_dir = "ants_image" //定位存放照片的文件名//
img_path = os.listdir(os.path.join(root_dir, target_dir)) //括号内先把照片路径整合为字符串，然后通过listdir把照片名称们变为列表赋给img_path//
label = target_dir.split('_')[0] // .split的用法是把字符串分割，从_出分割，形成两个新列表，列表的第一项，也就是“ants”赋给lable//
out_dir = 'ants_lable' /把''ants——lable“字符串给out_dir。方便一会把标签输出同一个文件夹下的新文件夹//
for i in img_path: //遍历img_path，把每一个照片名称以后缀名分割，【0】为照片编号//
file_name = i.split('.jpg')[0] //先看opon的括号内。。连接根目录、输出目录、文件名、加上tix后缀。这是with open的一个用法// //with open(file_path, mode) as file_variable:////其中，file_path是文件的路径，可以是相对路径或绝对路径；mode是打开文件的模式，例如"r"表示只读模式，"w"表示写入模式，"a"表示追加模式等。//
# 在这里执行对文件的操作
with open(os.path.join(root_dir,out_dir,"{}.txt".format(file_name)),'w+') as f:
f.write(label) //对每一个文件把他们的lable写进去
可见.split从_处把 ants_images分割成立两个列表，【0】是ants 【1】是img
import os
root_dir ="D:\\PyCharm\\Py_Projects\\XiaoTuDui\\datasets_jinjieP7\\train"
target_dir = "ants_image"
img_path = os.listdir(os.path.join(root_dir, target_dir))
label = target_dir.split('_')[0]
out_dir = 'ants_lable'
for i in img_path:
file_name = i.split('.jpg')[0]
with open(os.path.join(root_dir,out_dir,"{}.txt".format(file_name)),'w+') as f:
f.write(label)