/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

2 votes113 views1 answer
Flask import jsonify
Python
0 vote137 views1 answer
The grandest staircase of them all foobar solution
Python
1 vote113 views1 answer
Best Python Free Tutorial
Python
0 vote145 views1 answer
Read a csv and plot in python
Python
2 votes117 views1 answer
Standardize columns in pandas
Python
3 votes116 views2 answers
Python append row to 2d array
Python
34 votes101 views1 answer
Python main file with class
Python
0 vote117 views1 answer
Pandas make new dataframe
Python
0 vote110 views1 answer
Pylatex subsection
Python
0 vote114 views1 answer
Jupyter notebook make new lines
Python