- 博客(8)
- 收藏
- 关注
RSS订阅原创 tensorflow通过tf.keras.preprocessing.image.ImageDataGenerator创建数据集
这种方式可以直接读取分类文件夹里面的内容，并根据文件夹的名称生成相应的数据集generator.flow_from_directory
2023-04-05 14:59:56
242
原创 pytorch创建dataset的第二种方式
第二种方式是通过自定义dataset类，读取数据，然后区分训练数据和测试数据自定义类有两个要点，第一个要点是继承的父类是torch.utils.data.Dataset，第二个要点是要重写collate_fn并且在DataLoader中指定collate_fn=LEI.collate_fn
2023-04-05 14:31:09
282
原创 pytorch创建dataset的两种或方式
在pytorch中创建dataset的两种方式，第一种方式是通过torchvision.dataset.ImageFolder读取数据，然后再利用torch.utils.data.DataLoader第二种方式是继承torch.utils.data.Dataset类，重写__getitem()__和__len__以及静态方法collate_fn，其中collate_fn还需要通过from torch.utils.data.dataloader import default_collate进行处理
2023-04-05 14:29:41
708
4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人