python - 在我的情况下如何正确编写 collat​​e_fn？

Question

拜托，你能帮我找到解决我问题的方法吗？我想写 collat​​e_fn 使我的图片大小相等，但我不知道如何正确实现它。

Colab：链接

代码：

import pandas as pd
import numpy as np
from PIL import Image

from torchvision import transforms
from torch.utils.data.dataset import Dataset  # For custom datasets


class CustomDataset(Dataset):
    def __init__(self, csv_path):
        self.to_tensor = transforms.ToTensor()
        self.data_info = csv_path
        # First column contains the image paths
        self.image_arr = np.asarray(self.data_info.iloc[:, 0])
        # Second column is the labels
        self.label_arr = np.asarray(self.data_info.iloc[:, 1])
        # Calculate len
        self.data_len = len(self.data_info.index)

    def __getitem__(self, index):
        # Get image name from the pandas df
        single_image_name = self.image_arr[index]
        # Open image        
        IMAGE_SIZE = [224,224]

        response = requests.get(single_image_name)
        img_as_img = Image.open(BytesIO(response.content)).resize(IMAGE_SIZE)

        # Transform image to tensor
        img_as_tensor = self.to_tensor(img_as_img)

        # Get label(class) of the image based on the cropped pandas column
        single_image_label = self.label_arr[index]

        return (img_as_tensor, single_image_label)

    def __len__(self):
        return self.data_len

Answer 1

为了将图像大小调整为相同大小，您可以使用 opencv 库。为了安装库，请运行以下命令。

pip install opencv-python

您需要使用的功能如下。

cv2.resize(src, dsize[, dst[, fx[, fy[, interpolation]]]])

您可以在以下链接中找到该库的详细文档。