本文共 2399 字，大约阅读时间需要 7 分钟。

���������������������������������������������Tensor���������������������������������������������������������������������������PyTorch���������������������������������������

���������������������

���������������������������������

���������������

• ��������������������� Inherits torch.utils.data.Dataset ������������������������������
• ������������������������������������������������������������������������������������
• ������������������������������������������������������������������

���������������

������������������������������������

import os
import random
from PIL import Image
from torch.utils.data import Dataset

2. ������������������

������������������������������������������������������������������

import os
from PIL import Image
from torch.utils.data import Dataset
random.seed(1)  # ������������������������������������������
major_config = {
    'classes': ['dog', 'cat', 'car'],
    'class_label': {
        'dog': 0,
        'cat': 1,
        'car': 2
    }
}

3. ���������������

������������������������������������������������������������������

class LoadDataset(Dataset):
    def __init__(self, data_dir, transform=None):
        self.img_label_list = get_all_image_labels(data_dir)
        self.transform = transform

4. ���������������

������������������������������������������

def get_image_label_by_index(dataset, index):
    return dataset.img_label_list[index]

5. ������������������

������������������������������

def get_dataset_length(dataset):
    return len(dataset.img_label_list)

���������������

6. ������������������������������������������������������������������ PNG ��� JPG���������������������������

7. ���������������������������������������������������������������������������

8. ���������������������������������������������������One-hot encoding������������������������������

9. ���������������������������������������������������������������������

10. ���������������������������������������������������������������������������������������������������������

���������������������������������������������������������������������������������������������������������������������������������������