使用CSV文件作为标签格式的输入处理

CSV文件的读取。

import pandas as pd
df = pd.read_csv("train.csv")
print(df.head())   # 打印head和前几个数据

# 遍历获取值，则：
for data in df.values:
	print(data)  # 列表

import pandas as pd
df = pd.read_csv("train.csv")
data = df.reset_index()  # 给数据建立索引
print(data.loc[0])  # 读取第0个数据
print(data.loc[0, "images"])  # 读取第0个数据的head为images的数据

class MyDataset(Dataset):
	def __init__(self, csv_data, transform):
		self.data = pd.read_csv(csv_data).reset_index()
		self.transform = transfrom
	def __len__(self):
		return len(self.data)
	def __getitem__(self, idx):
		img_name = self.data.loc[idx, "image"]
		img = Image.open(img_name)
		img = self.transform(img)
		label = torch.tensor(self.data.loc[idx, "label"])
		return {"image": image, "labels": label}