如何将文件中的图片按照标签做成表格数据用于可视化使用
import os
import torch
from torch import nn
import warnings
import numpy as np
import cv2
import pandas as pd
from torch.utils.data import DataLoader
from torchvision import datasets, transforms
transform = transforms.Compose([
transforms.ToTensor(),
transforms.RandomResizedCrop(size=256, scale=(0.8, 1)),
transforms.CenterCrop((224, 224)),
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))])
test_data = datasets.ImageFolder(root="test", transform=transform)
print("测试集长度", len(test_data))
print(("classes_number", len(test_data.classes)))
print(("classes_names", test_data.classes))
test_loader = DataLoader(test_data, batch_size=1, shuffle=False, num_workers=0)
print("测试集的总批次数", len(test_loader))
img_paths = [each[0] for each in test_data.imgs]
df = pd.DataFrame()
df["picture_path"] = img_paths
df["picture_labels"] = test_data.targets
df.to_csv("mini.csv", index=False)
print(df)