/Python/Pytorch dataloader to numpy array

Pytorch dataloader to numpy array

# index is the specific data your want in dataloader
import numpy as np
def transfer_dataloader(dataloader, index=0):
    cache_list = list(iter(dataloader))
    assert len(cache_list) > 0
    assert index < len(cache_list[0])
    result_list = np.array(list(map(lambda x: x[index].numpy(), cache_list)))
    return result_list

Similar codes

0 vote292 views1 answer
Python catch output to terminal
Python
14 votes289 views5 answers
How to read a excel file in python
Python
1 vote277 views1 answer
Python 2 print sep end
Python
3 votes254 views1 answer
Python how to find gcd
Python
2 votes263 views3 answers
How to print dataframe in python without index
Python
6 votes270 views4 answers
Openpyxl
Python
0 vote264 views1 answer
How to create a graphical user interface in python
Python
1 vote242 views1 answer
Why do we use graphs?
Python
17 votes288 views4 answers
How to check how many items are in a list python
Python
0 vote237 views1 answer
Python continue inner for loop
Python