加载数据集MyDataset.py
数据形式:
标签格式:
猫:1,0
狗:0,1
from torch.utils.data import Dataset,DataLoader
import os
from PIL import Image
import torch
import numpy as np
from to
加载数据集MyDataset.py
数据形式:
标签格式:
猫:1,0
狗:0,1
from torch.utils.data import Dataset,DataLoader
import os
from PIL import Image
import torch
import numpy as np
from to