/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 vote236 views2 answers
Python parameter with default value and type
Python
6 votes270 views3 answers
Flask cors
Python Flask
3 votes243 views2 answers
Python use variable in another file
Python
0 vote232 views1 answer
Drop portion of string in dataframe python
Python
6 votes265 views1 answer
Find and replace string dataframe
Python
1 vote243 views1 answer
Libreoffice add line in table
Python
2 votes273 views1 answer
Matplotlib savefig not working
Python
0 vote274 views1 answer
Check correlation of each column with the target in python
Python
0 vote242 views1 answer
Login views django template passing
Python
1 vote284 views1 answer
Geometric progression in python
Python