pytorch：关于enumerate，Dataset和Dataloader_enumerate(data_loader)

作者：Monodyee | 2024-02-17 22:22:46

踩

enumerate(data_loader)

数据读取机制
在对Dataloader进行读取时，通常使用enumerate函数，enumerate() 函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列，同时列出数据和数据下标。
例如：for k, batch in enumerate(loader):
其中，k代表下标值，batch代表可遍历的数据对象，通常，loader里包含很多个数据对象，那么我们应该怎么保证batch就是我们所需要的数据呢？
通过Dataset的定义可以实现我们需要的数据。Dataset是用来定义数据从哪里读取，以及如何读取的问题，通过重写Dataset抽象类的__getitem__()函数，getitem()函数有几个return返回值，batch就是相应长度的list。
例如：getitem()函数的return值为
return img, target
相应的batch为长度为2的list。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Monodyee/article/detail/101837