cuốn sách gpt4 ai đã làm

python - 如何准备我自己的数据集并使用 pytorch 或 tensorflow 对其进行转换?

In lại Tác giả: Vũ trụ không gian 更新时间:2023-11-03 21:22:47 26 4
mua khóa gpt4 Nike

实际上我想做的是,我有大约 80 张图片的小图像数据,然后我想正确准备它,所以我不确定我的步骤是否正确。

class MyCustomDataset(Dataset):
def __init__(self,path,transform=None):
self.transformations = transforms.Compose([transforms.Normalize((.5, .5, .5), (.5, .5, .5)),
transforms.ToTensor()])
self.name=os.listdir(path)
self.count=np.size(self.name)
self.transform = transform
self.to_tensor = transforms.ToTensor()
#transforms.CenterCrop(100),
def __getitem__(self, index):

label=self.name[index]


im3 = Image.open(path2 + '//' + label)

imge=self.to_tensor(im3)
if self.transform is not None:
imge = self.transform(imge)
return (imge, label)

def __len__(self):
return self.count # of how many data(images?) you have

nếu __name__ == '__main__':
# Call the dataset

path=path2
custom_dataset = MyCustomDataset(path)#.(Dataset).__init__(path)
data_loader = torch.utils.data.DataLoader(custom_dataset, batch_size=100, shuffle=True)
num_batches = len(data_loader)

câu trả lời hay nhất

您的实现中有两个 transforms.ToTensor()。你只需要一个。请注意,某些变换对 PIL.Image 对象进行操作,而其他变换则对张量进行操作。请参阅torchvision.transforms哪个是哪个。

关于python - 如何准备我自己的数据集并使用 pytorch 或 tensorflow 对其进行转换?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54112169/

26 4 0
Bài viết khuyến nghị: opencv - 使用 OpenCV 计算 Jade 米树
Bài viết khuyến nghị: c# - MVC 区域路由
Bài viết khuyến nghị: css - 将 CSS 样式分组以备将来使用
Bài viết khuyến nghị: c# - 使用 Monitor.Wait 和 Monitor.Pulse 时出现内存不足异常
không gian vũ trụ
Hồ sơ

Tôi là một lập trình viên xuất sắc, rất giỏi!

Nhận phiếu giảm giá taxi Didi miễn phí
Phiếu giảm giá taxi Didi
Chứng chỉ ICP Bắc Kinh số 000000
Hợp tác quảng cáo: 1813099741@qq.com 6ren.com
Xem sitemap của VNExpress