当前位置：网站首页>Several ways of pytoch data encapsulation before entering the network

Several ways of pytoch data encapsulation before entering the network

2022-04-22 00:43:00 【Burn, man】

Pytorch Several ways of data encapsulation before entering the network

1、 utilize Pytorch Library method

To put it simply, use datasets.ImageFolder And torch.utils.data.DataLoader These two methods .
The data storage format needs to be as follows
Insert picture description here
Here is a code example
1、 Data processing 

train_transforms = transforms.Compose([
    transforms.Resize((64,

版权声明
本文为[Burn, man]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/04/202204211751447264.html

边栏推荐

猜你喜欢

随机推荐